Searched refs:axis_type (Results 1 – 2 of 2) sorted by relevance
164 DataType axis_type = input_type(2); in Compile() local165 OP_REQUIRES(context, axis_type == DT_INT32 || axis_type == DT_INT64, in Compile()
123 #define TF_LITE_ARG_MIN_MAX(data_type, axis_type, output_type) \ in Eval() argument126 GetTensorData<axis_type>(axis), GetTensorShape(output), \ in Eval()