Searched refs:cluster_attr (Results 1 – 6 of 6) sorted by relevance
26 }) {cluster_attr = "cluster_attr"} : () -> ()62 }) {cluster_attr = "cluster_attr"} : () -> ()91 }) {cluster_attr = "cluster_attr"} : () -> ()120 }) {cluster_attr = "cluster_attr"} : () -> ()148 }) {cluster_attr = "cluster_attr"} : () -> ()176 }) {cluster_attr = "cluster_attr"} : () -> ()213 }) {cluster_attr = "cluster_attr"} : () -> ()249 }) {cluster_attr = "cluster_attr"} : () -> ()284 }) {cluster_attr = "cluster_attr"} : () -> ()326 }) {cluster_attr = "cluster_attr"} : () -> (tensor<f32>)[all …]
15 // CHECK: {cluster_attr = "cluster_attr"}22 }) {cluster_attr = "cluster_attr"} : () -> tensor<*xi32>40 // CHECK: {cluster_attr = "cluster_attr"}48 }) {cluster_attr = "cluster_attr"} : () -> tensor<*xi32>68 // CHECK: {cluster_attr = "cluster_attr"}77 }) {cluster_attr = "cluster_attr"} : () -> tensor<*xi32>99 // CHECK: {cluster_attr = "cluster_attr"}109 }) {cluster_attr = "cluster_attr"} : () -> tensor<1xi32>139 }) {cluster_attr = "cluster_attr"} : () -> tensor<*xi32>167 // CHECK: {cluster_attr = "cluster_attr"} : () -> tensor<f32>[all …]
17 // CHECK: {_tpu_replicate = "x", cluster_attr = "cluster_attr", device = "y"}18 }) {cluster_attr = "cluster_attr", _tpu_replicate = "x", device = "y"} : () -> tensor<f32>
370 }) {cluster_attr = "cluster_attr"} : () -> ()385 }) {cluster_attr = "cluster_attr"} : () -> ()398 }) {cluster_attr = "cluster_attr"} : () -> ()413 }) {cluster_attr = "cluster_attr"} : () -> ()428 }) {cluster_attr = "cluster_attr"} : () -> ()443 }) {cluster_attr = "cluster_attr"} : () -> ()466 }) {cluster_attr = "cluster_attr"} : () -> ()492 }) {cluster_attr = "cluster_attr"} : () -> ()513 }) {cluster_attr = "cluster_attr"} : () -> ()
114 }) {cluster_attr = "cluster_attr"} : () -> tensor<?xi32>119 // CHECK-SAME: cluster_attr = "cluster_attr"
324 const AttrValue* cluster_attr = node->attrs().Find(kTPUReplicateAttr); in FindTaggedNodes() local325 if (cluster_attr == nullptr) { in FindTaggedNodes()329 const string& cluster = cluster_attr->s(); in FindTaggedNodes()350 const AttrValue* cluster_attr = node->attrs().Find(kTPUReplicateAttr); in FindTaggedNodes() local353 if (cluster_attr == nullptr) { in FindTaggedNodes()360 const string& cluster = cluster_attr->s(); in FindTaggedNodes()