Searched refs:weight_slice_shape_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 187 &builder_, weight_transposed_, weight_slice_shape_, begin_i2c_values, in SetWeightForInputToCellGate()
188 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToCellGate()
197 weight_slice_shape_, begin_i2i_values, in SetWeightForInputToInputGate()
198 weight_slice_shape_, in SetWeightForInputToInputGate()
207 &builder_, weight_transposed_, weight_slice_shape_, begin_i2f_values, in SetWeightForInputToForgetGate()
208 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToForgetGate()
217 &builder_, weight_transposed_, weight_slice_shape_, begin_i2o_values, in SetWeightForInputToOutputGate()
218 weight_slice_shape_, weight_slice_size_input_values_, in SetWeightForInputToOutputGate()
225 &builder_, weight_transposed_, weight_slice_shape_, begin_rec2c_values, in SetWeightForRecurrentToCellGate()
226 weight_slice_shape_, weight_slice_size_recurrent_values_, in SetWeightForRecurrentToCellGate()
[all …]
|
| D | lstm_utils.h | 160 SmallVector<int64_t, 2> weight_slice_shape_; variable
|