Searched refs:_TPU_REPLICATE_ATTR (Results 1 – 3 of 3) sorted by relevance
140 node.attr[tpu._TPU_REPLICATE_ATTR].s = b"0"155 tpu._TPU_REPLICATE_ATTR)158 tpu._TPU_REPLICATE_ATTR)165 tpu._TPU_REPLICATE_ATTR)170 tpu._TPU_REPLICATE_ATTR)
94 _TPU_REPLICATE_ATTR = "_tpu_replicate" variable586 if (_TPU_REPLICATE_ATTR in op.node_def.attr and590 op._set_attr_with_buf(_TPU_REPLICATE_ATTR,2270 op._clear_attr(_TPU_REPLICATE_ATTR) # pylint: disable=protected-access
473 return tpu._TPU_REPLICATE_ATTR not in op.node_def.attr