Home
last modified time | relevance | path

Searched refs:fused_func_op_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dlstm_utils.cc189 fused_func_op_.getLoc()); in SetWeightForInputToCellGate()
200 fused_func_op_.getLoc()); in SetWeightForInputToInputGate()
209 fused_func_op_.getLoc()); in SetWeightForInputToForgetGate()
219 fused_func_op_.getLoc()); in SetWeightForInputToOutputGate()
227 fused_func_op_.getLoc()); in SetWeightForRecurrentToCellGate()
238 fused_func_op_.getLoc()); in SetWeightForRecurrentToInputGate()
247 fused_func_op_.getLoc()); in SetWeightForRecurrentToForgetGate()
256 fused_func_op_.getLoc()); in SetWeightForRecurrentToOutputGate()
263 bias_size_values_, fused_func_op_.getLoc()); in SetBiasToCellGate()
273 bias_size_values_, fused_func_op_.getLoc()); in SetBiasToInputGate()
[all …]
Dlstm_utils.h51 : fused_func_op_(fused_func_op), in ConvertLSTMCellSimpleToFusedLSTM()
104 FuncOp fused_func_op_; variable