Searched refs:weight_transposed_ (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()
196 : SliceRankedTensor(&builder_, weight_transposed_, in SetWeightForInputToInputGate()
207 &builder_, weight_transposed_, weight_slice_shape_, begin_i2f_values, in SetWeightForInputToForgetGate()
217 &builder_, weight_transposed_, weight_slice_shape_, begin_i2o_values, in SetWeightForInputToOutputGate()
225 &builder_, weight_transposed_, weight_slice_shape_, begin_rec2c_values, in SetWeightForRecurrentToCellGate()
234 : SliceRankedTensor(&builder_, weight_transposed_, in SetWeightForRecurrentToInputGate()
245 &builder_, weight_transposed_, weight_slice_shape_, begin_rec2f_values, in SetWeightForRecurrentToForgetGate()
254 &builder_, weight_transposed_, weight_slice_shape_, begin_rec2o_values, in SetWeightForRecurrentToOutputGate()
341 weight_transposed_ = in GenerateFusedOpOperands()
|
| D | lstm_utils.h | 112 Value weight_transposed_; variable
|