Home
last modified time | relevance | path

Searched defs:DynamicRaggedShape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Ddynamic_ragged_shape.py160 class DynamicRaggedShape(extension_type.BatchableExtensionType): class
989 def _merge_with(self, other: "DynamicRaggedShape") -> "DynamicRaggedShape":
1596 def broadcast_dynamic_shape(shape_x: DynamicRaggedShape,
1597 shape_y: DynamicRaggedShape) -> DynamicRaggedShape:
1623 def broadcast_to(rt_input, shape: DynamicRaggedShape):
1662 a: DynamicRaggedShape, b: DynamicRaggedShape
2829 def _get_broadcast_num_row_partitions(a: DynamicRaggedShape,
2830 b: DynamicRaggedShape):
2849 a: DynamicRaggedShape, b: DynamicRaggedShape, b_rps: Sequence[RowPartition],
2868 a: DynamicRaggedShape, b: DynamicRaggedShape
[all …]
H A Dragged_array_ops.py1079 shape: dynamic_ragged_shape.DynamicRaggedShape
1149 def ones(shape: dynamic_ragged_shape.DynamicRaggedShape,
1158 def zeros(shape: dynamic_ragged_shape.DynamicRaggedShape,
1167 def fill(dims: dynamic_ragged_shape.DynamicRaggedShape,
H A Ddynamic_ragged_shape_test.py138 x: DynamicRaggedShape,
139 y: DynamicRaggedShape,
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/
H A Dstructured_tensor.py111 ragged_shape: dynamic_ragged_shape.DynamicRaggedShape):
145 cls, ragged_shape: dynamic_ragged_shape.DynamicRaggedShape
1145 cls, shape: dynamic_ragged_shape.DynamicRaggedShape
1640 a: Optional[dynamic_ragged_shape.DynamicRaggedShape],
1641 b: Optional[dynamic_ragged_shape.DynamicRaggedShape]