Home
last modified time | relevance | path

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

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