Home
last modified time | relevance | path

Searched refs:scaling_factors_ptr (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsvdf.cc316 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
340 &scaling_factors_ptr[b]); in EvalHybrid()
341 scaling_factors_ptr[b] *= weights_feature_scale; in EvalHybrid()
351 scaling_factors_ptr, batch_size, &state->data.f[memory_size - 1], in EvalHybrid()
Dfully_connected.cc247 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
263 &unused_max, &scaling_factors_ptr[b]); in EvalHybrid()
265 scaling_factors_ptr[b] *= filter->params.scale; in EvalHybrid()
270 filter_data, num_units, input_size, quant_data, scaling_factors_ptr, in EvalHybrid()
Dunidirectional_sequence_rnn.cc245 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
261 quantized_input_ptr, quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
281 scaling_factors_ptr, hidden_state_ptr_batch, output_ptr_batch); in EvalHybrid()
Dbidirectional_sequence_rnn.cc479 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
506 fw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
530 bw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
557 fw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
583 bw_quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
Dbasic_rnn.cc207 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
213 quantized_input_ptr, quantized_hidden_state_ptr, scaling_factors_ptr, in EvalHybrid()
Dconv.cc603 float* scaling_factors_ptr = in EvalHybrid() local
612 &unused_min, &unused_max, &scaling_factors_ptr[b]); in EvalHybrid()
613 scaling_factors_ptr[b] *= filter->params.scale; in EvalHybrid()
651 op_params, scaling_factors_ptr, GetTensorShape(input), in EvalHybrid()
Dlstm_eval.cc1233 float* scaling_factors_ptr = scaling_factors->data.f; in EvalHybrid() local
1310 output_gate_scratch, scaling_factors_ptr, prod_scaling_factors_ptr, in EvalHybrid()
1369 scaling_factors_ptr, prod_scaling_factors_ptr, in EvalHybrid()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dneon_tensor_utils.cc212 const float* scaling_factors_ptr = scaling_factors + batch; in DotprodMatrixBatchFourVectorMultiplyAccumulate() local
303 [ scaling_factors_ptr ] "r"(scaling_factors_ptr), in DotprodMatrixBatchFourVectorMultiplyAccumulate()
Dlegacy_optimized_ops.h538 int pad_height, float* scaling_factors_ptr, in HybridConv() argument
552 HybridConv(op_params, scaling_factors_ptr, DimsToShape(input_dims), in HybridConv()
Doptimized_ops.h2283 inline void HybridConv(const ConvParams& params, float* scaling_factors_ptr, in HybridConv() argument
2348 scaling_factors_ptr[i] = scaling_factors_ptr[i / rows_per_batch]; in HybridConv()
2355 scaling_factors_ptr, /*n_batch=*/gemm_input_rows, output_data, in HybridConv()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2201 inline void HybridConv(const ConvParams& params, float* scaling_factors_ptr, in HybridConv() argument
2269 scaling_factors_ptr[i] = scaling_factors_ptr[i / rows_per_batch]; in HybridConv()
2276 scaling_factors_ptr, /*n_batch=*/gemm_input_rows, output_data, in HybridConv()
2290 int pad_height, float* scaling_factors_ptr, in HybridConv() argument
2304 HybridConv(op_params, scaling_factors_ptr, DimsToShape(input_dims), in HybridConv()