Searched defs:ArgMax (Results 1 – 9 of 9) 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 | 220 XlaOp ArgMax(XlaOp input, PrimitiveType output_type, int axis, bool stable, in ArgMax() function
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1157 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
1167 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
|
| D | legacy_reference_ops.h | 2131 void ArgMax(const T3* axis, const T1* input_data, in ArgMax() function
|
| /external/tensorflow/tensorflow/lite/toco/tflite/ |
| D | operator.cc | 1259 class ArgMax : public BuiltinOperator<ArgMaxOperator, ::tflite::ArgMaxOptions, class
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 1044 TEST_F(OpTest, ArgMax) { in TEST_F() argument
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 8079 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
8090 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
|
| D | legacy_optimized_ops.h | 4972 void ArgMax(const T3* axis, const T1* input_data, in ArgMax() function
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 22574 func ArgMax(scope *Scope, input tf.Output, dimension tf.Output, optional ...ArgMaxAttr) (output tf.… func
|