Searched refs:_create_distributed_tensor_spec (Results 1 – 1 of 1) sorted by relevance
1037 self._element_spec = _create_distributed_tensor_spec(1266 self._element_spec = _create_distributed_tensor_spec(2104 def _create_distributed_tensor_spec(strategy, tensor_spec): function