Searched refs:LoadTPUEmbeddingRMSPropParametersGradAccumDebugAttr (Results 1 – 1 of 1) sorted by relevance
31630 type LoadTPUEmbeddingRMSPropParametersGradAccumDebugAttr func(optionalAttr) type31636 …MSPropParametersGradAccumDebugTableId(value int64) LoadTPUEmbeddingRMSPropParametersGradAccumDebug…31644 …ropParametersGradAccumDebugTableName(value string) LoadTPUEmbeddingRMSPropParametersGradAccumDebug…31667 …put, num_shards int64, shard_id int64, optional ...LoadTPUEmbeddingRMSPropParametersGradAccumDebug… argument