Home
last modified time | relevance | path

Searched defs:fusion_op (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/_inductor/fx_passes/
Dmkldnn_fusion.py473 fusion_op, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1792 auto fusion_op = mlir::cast<mlir::lmhlo::FusionOp>(op); in EmitFusion() local
5149 mlir::lmhlo::FusionOp fusion_op, const HloComputation* fused_computation) { in EmitDynamicUpdateSlice()
5195 Status IrEmitterUnnested::EmitScatter(mlir::lmhlo::FusionOp fusion_op, in EmitScatter()