Searched refs:RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
32789 type RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAccumDebugAttr func(optionalAttr) type32793 …rametersGradAccumDebugTableId(value int64) RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAcc…32801 …etersGradAccumDebugTableName(value string) RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAcc…32809 …rametersGradAccumDebugConfig(value string) RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAcc…32828 …_shards int64, shard_id int64, optional ...RetrieveTPUEmbeddingFrequencyEstimatorParametersGradAcc… argument