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.

bool tensorflow::ops::MaxPoolWithArgmax::Attrs::include_batch_in_index_ = false

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

TF_MUST_USE_RESULT Attrs tensorflow::ops::MaxPoolWithArgmax::Attrs::Targmax(
  DataType x
)

Defaults to DT_INT64.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2021-11-15 UTC.