Searched defs:tpu_embedding_config (Results 1 – 8 of 8) sorted by relevance
77 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in TEST() local136 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in TEST() local170 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in TEST() local246 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in TEST() local
25 const tensorflow::tpu::TPUEmbeddingConfiguration& tpu_embedding_config) { in ComputeTotalTagCountForDynamicLearningRates()
512 TpuSerializedProto tpu_embedding_config; member565 TpuSerializedProto tpu_embedding_config; member
65 tensorflow::tpu::TPUEmbeddingConfiguration* tpu_embedding_config, in GetTPUEmbeddingConfiguration()106 const tensorflow::tpu::TPUEmbeddingConfiguration& tpu_embedding_config) { in ValidateEmbeddingTableNames()186 const tensorflow::tpu::TPUEmbeddingConfiguration& tpu_embedding_config, in GetLoadOrRetrieveNodesByTable()280 const tensorflow::tpu::TPUEmbeddingConfiguration& tpu_embedding_config, in CombinePerTableParametersForLoad()379 tensorflow::tpu::TPUEmbeddingConfiguration tpu_embedding_config; in Run() local
43 const tpu::TPUEmbeddingConfiguration& tpu_embedding_config) { in CheckNumInputsOrOutputs()112 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in MakeRecvActivationsNodeDef() local169 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in MakeSendGradientsNodeDef() local
247 tpu::TPUEmbeddingConfiguration tpu_embedding_config; in Run() local
75 def tpu_embedding_config(): function
228 def tpu_embedding_config(): function