Home
last modified time | relevance | path

Searched refs:DynamicRaggedShape (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Ddynamic_ragged_shape_test.py44 from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
122 shape = DynamicRaggedShape.from_lengths(lengths)
138 x: DynamicRaggedShape, argument
139 y: DynamicRaggedShape, argument
141 assert isinstance(x, DynamicRaggedShape)
142 assert isinstance(y, DynamicRaggedShape)
218 shape = DynamicRaggedShape.from_tensor(value)
220 expected = DynamicRaggedShape(row_partitions, inner_shape)
295 shape = DynamicRaggedShape.from_lengths(
297 expected = DynamicRaggedShape(row_partitions, inner_shape)
[all …]
Ddynamic_ragged_shape.py57 return DynamicRaggedShape.Spec(
63 return DynamicRaggedShape.Spec._from_tensor_shape([None], # pylint:disable=protected-access
72 return DynamicRaggedShape.Spec(
78 return DynamicRaggedShape.Spec(
114 return DynamicRaggedShape.Spec(
120 return DynamicRaggedShape.Spec(
127 return DynamicRaggedShape.from_tensor(encoding, dtype=spec.dtype)
160 class DynamicRaggedShape(extension_type.BatchableExtensionType): class
389 return DynamicRaggedShape([], [], dtype=dtype)
401 return DynamicRaggedShape(
[all …]
Dragged_array_ops.py985 value_shape = dynamic_ragged_shape.DynamicRaggedShape.from_tensor(value)
1065 if isinstance(shape, dynamic_ragged_shape.DynamicRaggedShape):
1079 shape: dynamic_ragged_shape.DynamicRaggedShape argument
1104 out_type=dtypes.int32) -> dynamic_ragged_shape.DynamicRaggedShape:
1116 return dynamic_ragged_shape.DynamicRaggedShape.from_tensor(input, out_type)
1123 ) -> dynamic_ragged_shape.DynamicRaggedShape:
1141 if not isinstance(shape_x, dynamic_ragged_shape.DynamicRaggedShape):
1142 shape_x = dynamic_ragged_shape.DynamicRaggedShape([], shape_x)
1143 if not isinstance(shape_y, dynamic_ragged_shape.DynamicRaggedShape):
1144 shape_y = dynamic_ragged_shape.DynamicRaggedShape([], shape_y)
[all …]
Dragged_dispatch_test.py42 from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
1065 'shape': DynamicRaggedShape.from_lengths([3, (1, 0, 2)]),
1072 'shape': DynamicRaggedShape.from_lengths([3, (1, 0, 3)]),
1086 'shape': DynamicRaggedShape.from_lengths([3, (1, 0, 2)])
1095 expected=lambda: DynamicRaggedShape.from_lengths([2, (2, 1)])),
1102 expected=lambda: DynamicRaggedShape.from_lengths([2, (2, 1)])),
1106 'shape_x': DynamicRaggedShape.from_lengths([2, (2, 3), 1]),
1107 'shape_y': DynamicRaggedShape.from_lengths([5])
1109 expected=lambda: DynamicRaggedShape.from_lengths([2, (2, 3), 5])),
1113 'shape_x': DynamicRaggedShape.from_lengths([2, (2, 3), 1]),
[all …]
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor_dynamic.py39 return dynamic_ragged_shape.DynamicRaggedShape._from_inner_shape(
48 return dynamic_ragged_shape.DynamicRaggedShape._from_inner_shape(
51 return dynamic_ragged_shape.DynamicRaggedShape.from_row_partitions(
Dstructured_tensor.py87 _ragged_shape: dynamic_ragged_shape.DynamicRaggedShape
111 ragged_shape: dynamic_ragged_shape.DynamicRaggedShape): argument
145 cls, ragged_shape: dynamic_ragged_shape.DynamicRaggedShape argument
1145 cls, shape: dynamic_ragged_shape.DynamicRaggedShape argument
1571 spec: Union[dynamic_ragged_shape.DynamicRaggedShape.Spec,
1574 ) -> dynamic_ragged_shape.DynamicRaggedShape.Spec:
1578 …return dynamic_ragged_shape.DynamicRaggedShape.Spec._from_spec(spec) # pylint: disable=protected-…
1622 field, dtype=None) -> dynamic_ragged_shape.DynamicRaggedShape:
1629 return dynamic_ragged_shape.DynamicRaggedShape(
1632 elif isinstance(shape, dynamic_ragged_shape.DynamicRaggedShape):
[all …]
Dstructured_tensor_spec_test.py31 from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
83 _ragged_shape=DynamicRaggedShape.Spec(
93 _ragged_shape=DynamicRaggedShape.Spec(
112 _ragged_shape=DynamicRaggedShape.Spec(
120 _ragged_shape=DynamicRaggedShape.Spec(
143 spec = StructuredTensor.Spec(_ragged_shape=DynamicRaggedShape.Spec(
Dstructured_tensor_test.py35 from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
773 _ragged_shape=DynamicRaggedShape.Spec(
885 "type_spec": StructuredTensor.Spec._from_shape(DynamicRaggedShape.Spec(
896 DynamicRaggedShape.Spec(
901 DynamicRaggedShape.from_lengths([1, 0]))
1047 DynamicRaggedShape.Spec(
1724 spec = StructuredTensor.Spec._from_shape(DynamicRaggedShape.Spec(
1729 spec = StructuredTensor.Spec._from_shape(DynamicRaggedShape.Spec(
Dstructured_array_ops.py35 name=None) -> dynamic_ragged_shape.DynamicRaggedShape:
43 out_type=dtypes.int32) -> dynamic_ragged_shape.DynamicRaggedShape:
Dstructured_array_ops_test.py31 from tensorflow.python.ops.ragged.dynamic_ragged_shape import DynamicRaggedShape
949 shape = DynamicRaggedShape._from_inner_shape([4], dtype=dtypes.int32)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.experimental.-dynamic-ragged-shape.pbtxt1 path: "tensorflow.experimental.DynamicRaggedShape"
3 is_instance: "<class \'tensorflow.python.ops.ragged.dynamic_ragged_shape.DynamicRaggedShape\'>"
Dtensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt1 path: "tensorflow.experimental.DynamicRaggedShape.Spec"
3 …is_instance: "<class \'tensorflow.python.ops.ragged.dynamic_ragged_shape.DynamicRaggedShape.Spec\'…
Dtensorflow.experimental.pbtxt8 name: "DynamicRaggedShape"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.experimental.-dynamic-ragged-shape.pbtxt1 path: "tensorflow.experimental.DynamicRaggedShape"
3 is_instance: "<class \'tensorflow.python.ops.ragged.dynamic_ragged_shape.DynamicRaggedShape\'>"
Dtensorflow.experimental.-dynamic-ragged-shape.-spec.pbtxt1 path: "tensorflow.experimental.DynamicRaggedShape.Spec"
3 …is_instance: "<class \'tensorflow.python.ops.ragged.dynamic_ragged_shape.DynamicRaggedShape.Spec\'…
Dtensorflow.experimental.pbtxt8 name: "DynamicRaggedShape"
/external/tensorflow/
DRELEASE.md332 … * Introduced `tf.experimental.DynamicRaggedShape`, which represents the shape of a RaggedTensor.