Searched refs:bias_slice_shape_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 261 bias2cell_ = SliceRankedTensor(&builder_, bias_, bias_slice_shape_, in SetBiasToCellGate()
262 begin_bias2c_values, bias_slice_shape_, in SetBiasToCellGate()
271 : SliceRankedTensor(&builder_, bias_, bias_slice_shape_, in SetBiasToInputGate()
272 begin_bias2i_values, bias_slice_shape_, in SetBiasToInputGate()
279 bias2forget_ = SliceRankedTensor(&builder_, bias_, bias_slice_shape_, in SetBiasToForgetGate()
280 begin_bias2f_values, bias_slice_shape_, in SetBiasToForgetGate()
288 bias2output_ = SliceRankedTensor(&builder_, bias_, bias_slice_shape_, in SetBiasToOutputGate()
289 begin_bias2o_values, bias_slice_shape_, in SetBiasToOutputGate()
511 bias_slice_shape_ = {n_cell_}; in Initialize()
|
| D | lstm_utils.h | 158 SmallVector<int64_t, 1> bias_slice_shape_; variable
|