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