Searched refs:LoadTPUEmbeddingADAMParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
9603 type LoadTPUEmbeddingADAMParametersGradAccumDebugAttr func(optionalAttr) type9609 …ingADAMParametersGradAccumDebugTableId(value int64) LoadTPUEmbeddingADAMParametersGradAccumDebugAt…9617 …ADAMParametersGradAccumDebugTableName(value string) LoadTPUEmbeddingADAMParametersGradAccumDebugAt…9640 …tput, num_shards int64, shard_id int64, optional ...LoadTPUEmbeddingADAMParametersGradAccumDebugAt… argument