Home
last modified time | relevance | path

Searched refs:axis_type (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc164 DataType axis_type = input_type(2); in Compile() local
165 OP_REQUIRES(context, axis_type == DT_INT32 || axis_type == DT_INT64, in Compile()
/external/tensorflow/tensorflow/lite/kernels/
Darg_min_max.cc123 #define TF_LITE_ARG_MIN_MAX(data_type, axis_type, output_type) \ in Eval() argument
126 GetTensorData<axis_type>(axis), GetTensorShape(output), \ in Eval()