Home
last modified time | relevance | path

Searched refs:RaggedTensorDynamicShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_shape_test.py30 from tensorflow.python.ops.ragged.ragged_tensor_shape import RaggedTensorDynamicShape
39 assert isinstance(x, RaggedTensorDynamicShape)
40 assert isinstance(y, RaggedTensorDynamicShape)
92 shape = RaggedTensorDynamicShape.from_tensor(value)
93 expected = RaggedTensorDynamicShape.from_dim_sizes(expected_dim_sizes)
113 shape = RaggedTensorDynamicShape.from_dim_sizes(dim_sizes)
114 expected = RaggedTensorDynamicShape.from_dim_sizes(expected_dim_sizes)
303 original_shape = RaggedTensorDynamicShape.from_dim_sizes(original_dim_sizes)
304 bcast_shape = RaggedTensorDynamicShape.from_dim_sizes(broadcast_dim_sizes)
373 x_shape = RaggedTensorDynamicShape.from_dim_sizes(x_dims)
[all …]
Dragged_tensor_shape.py35 class RaggedTensorDynamicShape(object): class
157 return RaggedTensorDynamicShape(dim_sizes[:inner_split],
170 return RaggedTensorDynamicShape(
260 return RaggedTensorDynamicShape(partitioned_dims, self._inner_dim_sizes)
265 return RaggedTensorDynamicShape([], inner_dims)
331 return RaggedTensorDynamicShape(
383 return RaggedTensorDynamicShape(partitioned_sizes, inner_sizes)
396 return RaggedTensorDynamicShape(partitioned_sizes, inner_sizes)
405 return RaggedTensorDynamicShape(partitioned_sizes, inner_sizes)
420 if not isinstance(shape_x, RaggedTensorDynamicShape):
[all …]
Dragged_batch_gather_with_default_op.py150 params_shape = ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(
160 return ragged_tensor_shape.RaggedTensorDynamicShape(
163 return ragged_tensor_shape.RaggedTensorDynamicShape(
185 return ragged_tensor_shape.RaggedTensorDynamicShape(
Dragged_dispatch.py204 ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(x),
205 ragged_tensor_shape.RaggedTensorDynamicShape.from_tensor(y))