Searched refs:inner_dim_sizes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_tensor_shape.py | 85 def __init__(self, partitioned_dim_sizes, inner_dim_sizes): argument
102 (partitioned_dim_sizes, inner_dim_sizes)):
107 inner_dim_sizes = ragged_util.convert_to_int_tensor(
108 inner_dim_sizes, dtype=dtypes.int64, name='inner_dim_sizes')
121 inner_dim_sizes.shape.assert_has_rank(1)
124 self._inner_dim_sizes = inner_dim_sizes
216 def inner_dim_sizes(self): member in RaggedTensorDynamicShape
263 [array_ops.ones([dims_to_add], dtypes.int64), self.inner_dim_sizes],
333 array_ops.identity(self.inner_dim_sizes))
475 return array_ops.broadcast_to(rt_input, shape.inner_dim_sizes)
[all …]
|
| D | ragged_batch_gather_with_default_op.py | 165 array_ops.concat([params_shape.inner_dim_sizes[:-1], [1]], axis=0))
186 pad_dims, params_shape.inner_dim_sizes)
|
| D | ragged_tensor_shape_test.py | 50 self.assertAllEqual(x.inner_dim_sizes, y.inner_dim_sizes)
|