Home
last modified time | relevance | path

Searched refs:fused_op (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_ophint_func_op.cc87 Operation** fused_op) { in BuildUnidirectionalSequenceRnnOp() argument
114 *fused_op = builder->create<TFL::UnidirectionalSequenceRNNOp>( in BuildUnidirectionalSequenceRnnOp()
124 Operation** fused_op) { in BuildUnidirectionalSequenceLSTMOp() argument
188 *fused_op = builder->create<TFL::UnidirectionalSequenceLSTMOp>( in BuildUnidirectionalSequenceLSTMOp()
198 Operation* fused_op = nullptr; in ConvertTfLiteFusedOpIfAvailable() local
202 composite_func_op, call_op, builder, &fused_op); in ConvertTfLiteFusedOpIfAvailable()
204 call_op.replaceAllUsesWith(fused_op); in ConvertTfLiteFusedOpIfAvailable()
207 composite_func_op, call_op, builder, &fused_op); in ConvertTfLiteFusedOpIfAvailable()
210 if (call_output.getType() != fused_op->getResult(0).getType()) { in ConvertTfLiteFusedOpIfAvailable()
213 call_output.replaceAllUsesWith(fused_op->getResult(0)); in ConvertTfLiteFusedOpIfAvailable()
Dextract_ophint.cc314 void AggregateAndRewireOutputs(OpBuilder* builder, Operation* fused_op) { in AggregateAndRewireOutputs()
336 insert_loc, unpack_output_types, fused_op->getResult(output_index), in AggregateAndRewireOutputs()
349 fused_op->getResult(output_index)); in AggregateAndRewireOutputs()
355 fused_op->getResult(output_index)); in AggregateAndRewireOutputs()
640 Operation* fused_op = BuildFusedFuncOp( in ConvertOphintToStub() local
647 op->moveBefore(fused_op); in ConvertOphintToStub()
651 ophint_composite_op.AggregateAndRewireOutputs(builder, fused_op); in ConvertOphintToStub()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc917 NodeDef fused_op; in AddFusedContractionNode() local
918 fused_op.set_name(bias_add.name()); in AddFusedContractionNode()
919 fused_op.set_device(contraction.device()); in AddFusedContractionNode()
920 fused_op.add_input(contraction.input(0)); // 0: input in AddFusedContractionNode()
921 fused_op.add_input(contraction.input(1)); // 1: filter in AddFusedContractionNode()
922 fused_op.add_input(bias_add.input(1)); // 2: bias in AddFusedContractionNode()
925 fused_op.set_op(kFusedConv2D); in AddFusedContractionNode()
926 CopyConv2DAttributes(contraction, &fused_op); in AddFusedContractionNode()
928 fused_op.set_op(kFusedMatMul); in AddFusedContractionNode()
929 CopyMatMulAttributes(contraction, &fused_op); in AddFusedContractionNode()
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1723 for (auto& fused_op : fused_op_contexts) { in PredictFusedOp() local
1724 auto op_cost = PredictCosts(fused_op); in PredictFusedOp()