Searched refs:none_af (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | load_quantization_recipe.cc | 65 StringAttr none_af; member
102 auto l2_norm = builder->create<L2NormalizationOp>(loc, int16, in, none_af); in CreateLayerNorm()
103 auto add = builder->create<AddOp>(loc, int16, in, l2_norm, none_af); in CreateLayerNorm()
105 none_af, fc_format, keep_dims); in CreateLayerNorm()
113 none_af, fc_format, keep_dims); in CreateGate()
115 none_af, fc_format, keep_dims); in CreateGate()
120 cell.getValue().second, none_af); in CreateGate()
172 loc, int16, forget_gate->getResult(0), lstm.input_cell_state(), none_af); in LoadForLSTMOp()
174 loc, int16, input_gate->getResult(0), cell_gate->getResult(0), none_af); in LoadForLSTMOp()
176 input_cell_state.output(), none_af); in LoadForLSTMOp()
[all …]
|
| D | optimize.cc | 371 auto none_af = rewriter.getStringAttr("NONE"); in matchAndRewrite() local
373 rewriter.create<AddOp>(add_op.getLoc(), bias, constant_val, none_af) in matchAndRewrite()
381 auto none_af = rewriter.getStringAttr("NONE"); in matchAndRewrite() local
383 rewriter.create<AddOp>(add_op.getLoc(), bias, constant_val, none_af) in matchAndRewrite()
|