Searched refs:_add_data_for_ragged_tensor (Results 1 – 1 of 1) sorted by relevance
921 def _add_data_for_ragged_tensor(self, tensor, weight, row_splits, values, member in TPUEmbedding988 self._add_data_for_ragged_tensor(inp, weight, indices_or_row_splits,