Searched defs:axis_type (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | arg_min_max_test.cc | 33 ArgBaseOpModel(TensorType input_type, int axis_value, TensorType axis_type, in ArgBaseOpModel()
85 int axis_value, TensorType axis_type, bool constant_axis, in ArgMaxOpModel()
100 int axis_value, TensorType axis_type, bool constant_axis, in ArgMinOpModel()
|
| D | arg_min_max.cc | 139 #define TF_LITE_ARG_MIN_MAX(data_type, axis_type, output_type) \ in Eval() argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_op.cc | 169 DataType axis_type = context->input_type(2); in XlaGatherWithBatchDimsOpImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 1591 auto axis_type = in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1247 auto axis_type = RankedTensorType::get({}, getElementTypeOrSelf(axis_attr)); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 412 DataType axis_type; member
3989 auto axis_type = Choose<DataType>({DT_INT32, DT_INT64}); in TEST_F() local
|