Home
last modified time | relevance | path

Searched defs:all_reduce_op (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbef_thunk.cc102 if (auto all_reduce_op = mlir::dyn_cast<mlir::lmhlo::AllReduceOp>(*op)) { in BefThunk() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc264 mlir::Operation** add_op, mlir::TF::DTensorAllReduceOp* all_reduce_op, in CheckWhileLoopOptimizationCriteria()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_function_importer.cc996 auto all_reduce_op = func_builder->create<mlir::mhlo::AllReduceOp>( in ImportInstructionImpl() local