Home
last modified time | relevance | path

Searched refs:AddDTensorAllReduceCombineOptimization (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Ddtensor_mlir_passes_internal.cc24 void AddDTensorAllReduceCombineOptimization(mlir::OpPassManager* pm){} in AddDTensorAllReduceCombineOptimization() function
H A Ddtensor_mlir_passes_internal.h24 void AddDTensorAllReduceCombineOptimization(mlir::OpPassManager* pm);
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_mlir_passes.cc196 AddDTensorAllReduceCombineOptimization(pm); in CreateDTensorMLIRPass()