Searched defs:ArgMax (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | argmax_op.h | 29 struct ArgMax { struct
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | arithmetic.cc | 168 XlaOp ArgMax(XlaOp input, PrimitiveType output_type, int axis) { in ArgMax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | reference_ops.h | 752 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
762 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
|
| H A D | legacy_reference_ops.h | 2133 void ArgMax(const T3* axis, const T1* input_data, in ArgMax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/ |
| H A D | operator.cc | 1263 class ArgMax : public BuiltinOperator<ArgMaxOperator, ::tflite::ArgMaxOptions, class
|
| /aosp_15_r20/external/pytorch/test/distributions/ |
| H A D | test_distributions.py | 2104 class ArgMax: class
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | optimized_ops.h | 8029 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
8040 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax() function
|
| H A D | legacy_optimized_ops.h | 4974 void ArgMax(const T3* axis, const T1* input_data, in ArgMax() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | randomized_tests.cc | 1679 TEST_F(OpTest, ArgMax) { in TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | wrappers.go | 1233 func ArgMax(scope *Scope, input tf.Output, dimension tf.Output, optional ...ArgMaxAttr) (output tf.… func
|