Searched defs:RaggedTensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_embedding_for_serving.py | 177 ragged: ragged_tensor.RaggedTensor,
178 weights: ragged_tensor.RaggedTensor,
365 inp: ragged_tensor.RaggedTensor,
366 weight: Optional[ragged_tensor.RaggedTensor], table: tf_variables.Variable,
|
| H A D | tpu_embedding_v1.py | 352 self, inp: ragged_tensor.RaggedTensor,
353 weight: Optional[ragged_tensor.RaggedTensor],
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_image_ops.py | 32 def resize_images_v2(images: ragged_tensor.RaggedTensor,
50 def resize_images_v1(images: ragged_tensor.RaggedTensor,
|
| H A D | ragged_tensor.py | 60 class RaggedTensor(composite_tensor.CompositeTensor, class
|