Searched refs:CollectiveGatherAttr (Results 1 – 1 of 1) sorted by relevance
38619 type CollectiveGatherAttr func(optionalAttr) type38623 func CollectiveGatherCommunicationHint(value string) CollectiveGatherAttr {38631 func CollectiveGatherTimeoutSeconds(value float32) CollectiveGatherAttr {38638 … group_key int64, instance_key int64, shape tf.Shape, optional ...CollectiveGatherAttr) (data tf.O… argument