43 const circle::Tensor *input = kernel.
input1();
44 const circle::Tensor *output = kernel.
output();
51 switch (Tensor::element_type(input))
54 case DataType::FLOAT32:
56 luci_interpreter_pal::ArgMinMax(
58 kernels::getTensorData<float>(input_data), kernels::getTensorData<int32_t>(axis_data),
60 kernels::getTensorData<int32_t>(output_data), std::greater<float>());
65 assert(
false &&
"Unsupported ArgMax input type");