Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_embedding_configuration_utils.h29 absl::StatusOr<int32_t> ComputeTotalTagCountForDynamicLearningRates(
Dtpu_embedding_configuration_utils.cc24 absl::StatusOr<int32_t> ComputeTotalTagCountForDynamicLearningRates( in ComputeTotalTagCountForDynamicLearningRates() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dtpu_embedding_software_deduplication_rewrite_pass.cc194 tpu::ComputeTotalTagCountForDynamicLearningRates(tpu_embedding_config); in MakeSendGradientsNodeDef()