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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/cc/struct/tensorflow/ops/max-pool-with-argmax/attrs