Searched refs:input_gate_bias_ptr (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/custom_logging_ops/ |
D | lstm.cc | 60 const float* input_gate_bias_ptr, const float* forget_gate_bias_ptr, in LstmStepWithAuxInput() argument 87 tensor_utils::VectorBatchVectorAssign(input_gate_bias_ptr, n_cell, in LstmStepWithAuxInput() 211 tensor_utils::VectorBatchVectorAdd(input_gate_bias_ptr, n_cell, n_batch, in LstmStepWithAuxInput()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | lstm_eval.cc | 828 const float* input_gate_bias_ptr, const float* forget_gate_bias_ptr, in LstmStepFloat() argument 859 input_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepFloat() 1019 const float* input_gate_bias_ptr, const float* forget_gate_bias_ptr, in LstmStepHybrid() 1149 input_layer_norm_coefficients_ptr, input_gate_bias_ptr, n_batch, in LstmStepHybrid() 1361 const int32_t* input_gate_bias_ptr, const int32_t* forget_gate_bias_ptr, in LstmStepInteger8x8_16() 1416 input_gate_bias_ptr, layer_norm_input_scale_a, layer_norm_input_scale_b, in LstmStepInteger8x8_16() 1613 const int32_t* input_gate_bias_ptr, const int32_t* forget_gate_bias_ptr, in LstmStepInteger8x8_8()
|
D | lstm.cc | 562 int32_t* input_gate_bias_ptr = nullptr; in PopulateQuantizedLstmParams8x8_8() local 613 input_gate_bias_ptr = input_gate_bias->data.i32; in PopulateQuantizedLstmParams8x8_8()
|
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/builtin_logging_ops/ |
D | lstm.cc | 177 const float* input_gate_bias_ptr, const float* forget_gate_bias_ptr, in LstmStepCalibration() argument 210 input_gate_bias_ptr, n_batch, n_input, n_aux_input, n_output, n_cell, in LstmStepCalibration()
|