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