32 Tensor::element_type(kernel.
output()));
34 Tensor::num_elements(kernel.
output()));
49 switch (Tensor::element_type(kernel.
input()))
52 case DataType::FLOAT32:
54 const float *input_data_float = kernels::getTensorData<float>(input_data);
55 float *output_data_float = kernels::getTensorData<float>(output_data);
58 output_data_float =
const_cast<float *
>(input_data_float);
61 assert(output_data_float);
71 assert(
false &&
"Unsupported type");