Searched refs:_create_distributed_tensor_spec (Results 1 – 2 of 2) sorted by relevance
1081 if element_spec != _create_distributed_tensor_spec(1107 self._element_spec = _create_distributed_tensor_spec(1404 self._element_spec = _create_distributed_tensor_spec(2056 def _create_distributed_tensor_spec(strategy, tensor_spec): function
1476 input_lib._create_distributed_tensor_spec(self.strategy,