44 const circle::Tensor *input = kernel.
input1();
45 const circle::Tensor *output = kernel.
output();
52 switch (Tensor::element_type(input))
55 case DataType::FLOAT32:
57 luci_interpreter_pal::ArgMinMax(
59 kernels::getTensorData<float>(input_data), kernels::getTensorData<int32_t>(axis_data),
61 kernels::getTensorData<int32_t>(output_data), std::less<float>());
66 assert(
false &&
"Unsupported ArgMax input type");