31 Tensor::element_type(kernel.
output()));
33 Tensor::num_elements(kernel.
output()));
48 switch (Tensor::element_type(kernel.
input()))
51 case DataType::FLOAT32:
53 const float *input_data_float = kernels::getTensorData<float>(input_data);
54 float *output_data_float = kernels::getTensorData<float>(output_data);
57 output_data_float =
const_cast<float *
>(input_data_float);
60 assert(output_data_float);
70 assert(
false &&
"Unsupported type");