Searched refs:bias_data_int32 (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | legacy_optimized_ops.h | 384 const int32* bias_data_int32, const Dims<4>& bias_dims, int32 output_offset, in FullyConnected() argument 400 bias_data_int32, DimsToShape(output_dims), output_data, in FullyConnected() 847 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument 865 bias_data_int32, DimsToShape(prev_state_dims), prev_state_data_int16, in LstmCell()
|
D | optimized_ops.h | 1229 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected() argument 1270 bias_data_int32, output_multiplier, output_shift, output_shape, in FullyConnected() 1276 filter_offset, bias_shape, bias_data_int32, in FullyConnected() 1291 ColVectorMap bias_vector(bias_data_int32, output_depth); in FullyConnected() 3680 const int32* bias_data_int32, in LstmCell() argument 3770 bias_data_int32, accum_multiplier, accum_shift, in LstmCell() 3784 ColVectorMap bias_vector(bias_data_int32, fc_output_depth); in LstmCell()
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | optimized_ops.h | 1237 const int32* bias_data_int32, const RuntimeShape& output_shape, in FullyConnected() argument 1278 bias_data_int32, output_multiplier, output_shift, output_shape, in FullyConnected() 1284 filter_offset, bias_shape, bias_data_int32, in FullyConnected() 1299 ColVectorMap bias_vector(bias_data_int32, output_depth); in FullyConnected() 1325 const int32* bias_data_int32, const Dims<4>& bias_dims, int32 output_offset, in FullyConnected() argument 1341 bias_data_int32, DimsToShape(output_dims), output_data, in FullyConnected() 3796 const int32* bias_data_int32, in LstmCell() argument 3886 bias_data_int32, accum_multiplier, accum_shift, in LstmCell() 3900 ColVectorMap bias_vector(bias_data_int32, fc_output_depth); in LstmCell() 4082 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | legacy_reference_ops.h | 546 const Dims<4>& weights_dims, const int32* bias_data_int32, in LstmCell() argument 564 bias_data_int32, DimsToShape(prev_state_dims), prev_state_data_int16, in LstmCell()
|
D | reference_ops.h | 2043 const int32* bias_data_int32, in LstmCell() argument 2131 int32 accum = bias_data_int32[out_c]; in LstmCell()
|