Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device_util.h629 Status InsertFunctionForTPUEmbeddingCheckpoint(
H A Ddtensor_device_util.cc1089 Status InsertFunctionForTPUEmbeddingCheckpoint( in InsertFunctionForTPUEmbeddingCheckpoint() function
H A Ddtensor_device.cc1416 Status s = InsertFunctionForTPUEmbeddingCheckpoint( in LowerToSPMDFunction()