Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dcreate_dtensor_mlir_passes.h46 CreateDTensorAllReduceSumOptimization();
H A Ddtensor_mlir_passes.cc194 CreateDTensorAllReduceSumOptimization()); in CreateDTensorMLIRPass()
H A Ddtensor_allreduce_sum_optimization.cc514 CreateDTensorAllReduceSumOptimization() { in CreateDTensorAllReduceSumOptimization() function
H A DPasses.td48 let constructor = "CreateDTensorAllReduceSumOptimization()";