Searched defs:ArgMax (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | argmax_op.h | 29 struct ArgMax { struct
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 178 XlaOp ArgMax(XlaOp input, PrimitiveType output_type, int axis) { in ArgMax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1532 class ArgMax : public BuiltinOperator<ArgMaxOperator, ::tflite::ArgMaxOptions, class
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3911 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
3921 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
|
| D | legacy_reference_ops.h | 2034 void ArgMax(const T3* axis, const T1* input_data, in ArgMax() function
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1043 TEST_F(OpTest, ArgMax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 16692 func ArgMax(scope *Scope, input tf.Output, dimension tf.Output, optional ...ArgMaxAttr) (output tf.… func
|