Searched refs:fusion_op (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1875 auto fusion_op = mlir::cast<mlir::lmhlo::FusionOp>(mlir_input.op); in HandleFusion() local
1882 GetOrCreateSubComputationFromRegion(&fusion_op.region(), in HandleFusion()
1916 ComputeMaxUnrollFactor(fusion_op, hlo_module_config_); in HandleFusion()
1927 IrName(mlir::GetNameFromLoc(fusion_op.getLoc())), in HandleFusion()
1929 fusion_op, launch_dimensions.launch_bound(), &b_))); in HandleFusion()
2017 fusion_op, ir_emitter_context_->allocations())) { in HandleFusion()
2028 BuildKernelThunkForMlir(fusion_op, mlir_input.thunk_info, in HandleFusion()
2052 GetOrCreateSubComputationFromRegion(&fusion_op.region(), in HandleFusion()
5128 if (auto fusion_op = mlir::dyn_cast<mlir::lmhlo::FusionOp>(input.op)) { in CheckAndEmitHloWithTile021() local
5129 params_012 = FilterInputsForShmemTranspose(fusion_op, params_012); in CheckAndEmitHloWithTile021()
|