Missed TensorFlow World? Check out the recap. Learn more

tensorflow::ops::ArgMax::Attrs

#include <math_ops.h>

Optional attribute setters for ArgMax.

Summary

Public attributes

output_type_ = DT_INT64
DataType

Public functions

OutputType(DataType x)
TF_MUST_USE_RESULT Attrs
Defaults to DT_INT64.

Public attributes

output_type_

DataType tensorflow::ops::ArgMax::Attrs::output_type_ = DT_INT64

Public functions

OutputType

TF_MUST_USE_RESULT Attrs tensorflow::ops::ArgMax::Attrs::OutputType(
  DataType x
)

Defaults to DT_INT64.