tensorflow::ops::TopK::Attrs

#include <nn_ops.h>

Optional attribute setters for TopK.

Summary

Public attributes

index_type_ = DT_INT32
DataType
sorted_ = true
bool

Public functions

IndexType(DataType x)
TF_MUST_USE_RESULT Attrs
Defaults to DT_INT32.
Sorted(bool x)
TF_MUST_USE_RESULT Attrs
If true the resulting k elements will be sorted by the values in descending order.

Public attributes

index_type_

DataType tensorflow::ops::TopK::Attrs::index_type_ = DT_INT32

sorted_

bool tensorflow::ops::TopK::Attrs::sorted_ = true

Public functions

IndexType

TF_MUST_USE_RESULT Attrs tensorflow::ops::TopK::Attrs::IndexType(
  DataType x
)

Defaults to DT_INT32.

Sorted

TF_MUST_USE_RESULT Attrs tensorflow::ops::TopK::Attrs::Sorted(
  bool x
)

If true the resulting k elements will be sorted by the values in descending order.

Defaults to true