Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/utils/
H A Ddtensor_mlir_passes_internal.cc30 void AddDTensorEmbeddingCheckpointPass(mlir::OpPassManager* pm){} in AddDTensorEmbeddingCheckpointPass() function
H A Ddtensor_mlir_passes_internal.h30 void AddDTensorEmbeddingCheckpointPass(mlir::OpPassManager* pm);
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_mlir_passes.cc164 AddDTensorEmbeddingCheckpointPass(pm); in CreateDTensorMLIRPass()