Home
last modified time | relevance | path

Searched defs:restore_v2 (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dsave_restore_spmd_expander.cc662 mlir::TF::DTensorRestoreV2Op restore_v2 = in ExpandDTensorRestoreV2Op() local
849 mlir::TF::RestoreV2Op restore_v2 = mlir::cast<mlir::TF::RestoreV2Op>(op); in ComputeLayoutForward() local
882 mlir::TF::DTensorRestoreV2Op restore_v2 = in ComputeLayoutForward() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dgraph_properties_test.cc1043 Output restore_v2 = in TEST_F() local