Searched refs:_generateRaggedTensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_to_tensor_op_test.py | 779 rt_input = self._generateRaggedTensor(shape, ragged_rank, dtype, fill)
781 self._generateRaggedTensor(default_shape, 0, dtype), dtype=dtype)
829 def _generateRaggedTensor(self, shape, ragged_rank, dtype, fill=None, axis=0): member in RaggedToDenseBenchmark
842 self._generateRaggedTensor(shape, ragged_rank, dtype, fill, axis + 1)
|