Searched refs:_TPU_REPLICATE_ATTR (Results 1 – 5 of 5) sorted by relevance
112 node.attr[tpu._TPU_REPLICATE_ATTR].s = b"0"127 tpu._TPU_REPLICATE_ATTR)130 tpu._TPU_REPLICATE_ATTR)137 tpu._TPU_REPLICATE_ATTR)142 tpu._TPU_REPLICATE_ATTR)
71 _TPU_REPLICATE_ATTR = "_tpu_replicate" variable371 if _TPU_REPLICATE_ATTR in op.node_def.attr:373 op._set_attr(_TPU_REPLICATE_ATTR,1608 op._clear_attr(_TPU_REPLICATE_ATTR) # pylint: disable=protected-access
489 return tpu._TPU_REPLICATE_ATTR not in op.node_def.attr
3372 if tpu._TPU_REPLICATE_ATTR in c.op.node_def.attr: # pylint: disable=protected-access
24 from tensorflow.python.tpu.tpu import _TPU_REPLICATE_ATTR