Searched refs:_get_input_shape_for_tensor (Results 1 – 1 of 1) sorted by relevance
1348 self._get_input_shape_for_tensor(tensor, feature, path))1357 def _get_input_shape_for_tensor(self, tensor, feature, path) -> TensorShape: member in TPUEmbedding