Searched refs:CollectiveGather (Results 1 – 9 of 9) sorted by relevance
2 name: "CollectiveGather"43 name: "CollectiveGather"91 name: "CollectiveGather"
2 name: "CollectiveGather"43 name: "CollectiveGather"
2 graph_op_name: "CollectiveGather"
768 name: "CollectiveGather"
38638 func CollectiveGather(scope *Scope, input tf.Output, group_size int64, group_key int64, instance_ke… func
1870 def TF_CollectiveGatherOp : TF_Op<"CollectiveGather", []> {
7620 name: "CollectiveGather"
12963 name: "CollectiveGather"