Searched defs:axis_type (Results 1 – 5 of 5) 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 | 143 #define TF_LITE_ARG_MIN_MAX(data_type, axis_type, output_type) \ in Eval() argument
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | arg_min_max.cc | 56 #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 | 168 DataType axis_type = context->input_type(2); in XlaGatherWithBatchDimsOpImpl() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 1176 auto axis_type = mlir::RankedTensorType::get({}, rewriter.getIntegerType(64)); in matchAndRewrite() local
|