Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Drestore_shape_inference.cc159 CreateDTensorInferShapesForRestoreV2Op() { in CreateDTensorInferShapesForRestoreV2Op() function
H A Dcreate_dtensor_mlir_passes.h136 CreateDTensorInferShapesForRestoreV2Op();
H A Ddtensor_mlir_passes.cc154 pm->addPass(CreateDTensorInferShapesForRestoreV2Op()); in CreateDTensorMLIRPass()
H A DPasses.td310 let constructor = "CreateDTensorInferShapesForRestoreV2Op()";