tensorflow:: ops:: MaxPoolWithArgmax:: Attrs
#include <nn_ops.h>
Optional attribute setters for MaxPoolWithArgmax.
Summary
Public attributes |
|
---|---|
Targmax_ = DT_INT64
|
DataType
|
include_batch_in_index_ = false
|
bool
|
Public functions |
|
---|---|
IncludeBatchInIndex(bool x)
|
TF_MUST_USE_RESULT Attrs
Whether to include batch dimension in flattened index of
argmax . |
Targmax(DataType x)
|
TF_MUST_USE_RESULT Attrs
Defaults to DT_INT64.
|
Public attributes
Targmax_
DataType tensorflow::ops::MaxPoolWithArgmax::Attrs::Targmax_ = DT_INT64
include_batch_in_index_
bool tensorflow::ops::MaxPoolWithArgmax::Attrs::include_batch_in_index_ = false
Public functions
IncludeBatchInIndex
TF_MUST_USE_RESULT Attrs tensorflow::ops::MaxPoolWithArgmax::Attrs::IncludeBatchInIndex( bool x )
Whether to include batch dimension in flattened index of argmax
.
Defaults to false
Targmax
TF_MUST_USE_RESULT Attrs tensorflow::ops::MaxPoolWithArgmax::Attrs::Targmax( DataType x )
Defaults to DT_INT64.