Searched refs:_add_data_for_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_v2.py | 889 def _add_data_for_tensor(self, tensor, weight, indices, values, weights, member in TPUEmbedding
981 self._add_data_for_tensor(inp, weight, indices_or_row_splits, values,
|