| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv.cc | 845 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors_tensor); in EvalHybridPerChannel() local
856 quantized_input_ptr_batch + offset, &scaling_factors_ptr[b], in EvalHybridPerChannel()
881 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybridPerChannel()
900 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybridPerChannel()
941 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors_tensor); in EvalHybrid() local
951 &unused_min, &unused_max, &scaling_factors_ptr[b]); in EvalHybrid()
952 scaling_factors_ptr[b] *= filter->params.scale; in EvalHybrid()
973 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybrid()
|
| D | batch_matmul.cc | 464 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid() local
477 input_size, quant_data, scaling_factors_ptr, in EvalHybrid()
482 scaling_factors_ptr[b] *= filter->params.scale; in EvalHybrid()
493 filter_shape, filter_data, input_shape, quant_data, scaling_factors_ptr, in EvalHybrid()
499 filter_shape, filter_data, input_shape, quant_data, scaling_factors_ptr, in EvalHybrid()
|
| D | unidirectional_sequence_rnn.cc | 299 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid() local
324 quantized_input_ptr, quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
347 scaling_factors_ptr, hidden_state_ptr_batch, output_ptr_batch, in EvalHybrid()
|
| D | depthwise_conv.cc | 469 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors_tensor); in EvalHybridPerChannel() local
480 quantized_input_ptr_batch + offset, &scaling_factors_ptr[b], in EvalHybridPerChannel()
501 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybridPerChannel()
509 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybridPerChannel()
|
| D | bidirectional_sequence_rnn.cc | 575 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid() local
611 fw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
638 bw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
669 fw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
700 bw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
|
| D | fully_connected.cc | 482 float* scaling_factors_ptr = in EvalHybridImpl() local
494 quant_data, scaling_factors_ptr, in EvalHybridImpl()
499 scaling_factors_ptr[b] *= filter->params.scale; in EvalHybridImpl()
509 output_depth, input_depth, quant_data, scaling_factors_ptr, batch_size, in EvalHybridImpl()
513 filter_data, output_depth, input_depth, quant_data, scaling_factors_ptr, in EvalHybridImpl()
|
| D | basic_rnn.cc | 261 float* scaling_factors_ptr = GetTensorData<float>(scaling_factors); in EvalHybrid() local
273 quantized_input_ptr, quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | conv.h | 188 const ConvParams& params, float* scaling_factors_ptr, in HybridConvPerChannel() argument
248 acc * per_channel_scale[out_channel] * scaling_factors_ptr[batch]; in HybridConvPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | depthwise_conv.h | 209 const DepthwiseParams& params, float* scaling_factors_ptr, in DepthwiseConvHybridPerChannel() argument
271 per_channel_scale[output_channel] * scaling_factors_ptr[batch]; in DepthwiseConvHybridPerChannel()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | neon_tensor_utils.cc | 320 const float* scaling_factors_ptr = scaling_factors + batch; in DotprodMatrixBatchFourVectorMultiplyAccumulate() local
422 [scaling_factors_ptr] "r"(scaling_factors_ptr), in DotprodMatrixBatchFourVectorMultiplyAccumulate()
450 const float* scaling_factors_ptr = scaling_factors + batch; in DotprodMatrixBatchFourVectorMultiplyAccumulate() local
546 [scaling_factors_ptr] "r"(scaling_factors_ptr), in DotprodMatrixBatchFourVectorMultiplyAccumulate()
|
| D | optimized_ops.h | 1333 inline void HybridConv(const ConvParams& params, float* scaling_factors_ptr, in HybridConv() argument
1409 scaling_factors_ptr[i] = scaling_factors_ptr[i / rows_per_batch]; in HybridConv()
1418 scaling_factors_ptr, /*n_batch=*/gemm_input_rows, accum_scratch, in HybridConv()
1426 const ConvParams& params, float* scaling_factors_ptr, in HybridConvPerChannel() argument
1524 const float input_scale = scaling_factors_ptr[b]; in HybridConvPerChannel()
|
| D | legacy_optimized_ops.h | 2554 int pad_height, float* scaling_factors_ptr, in HybridConv() argument
2570 HybridConv(op_params, scaling_factors_ptr, DimsToShape(input_dims), in HybridConv()
|