Home
last modified time | relevance | path

Searched refs:nested_row_splits (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor.py563 def from_nested_row_splits(cls, flat_values, nested_row_splits, name=None): argument
583 if isinstance(nested_row_splits, ops.Tensor):
586 [flat_values] + list(nested_row_splits)):
588 for splits in reversed(nested_row_splits):
744 def nested_row_splits(self): member in RaggedTensor
969 nested_splits = self.nested_row_splits
1388 self.nested_row_splits, self.flat_values, name=name)
1428 for row_splits in reversed(self.nested_row_splits):
1472 return (self.flat_values,) + self.nested_row_splits
1562 value.nested_row_splits)
[all …]
Dragged_tensor_test.py157 nested_row_splits=([0, 3, 3, 5], [0, 4, 4, 7, 8, 8]))
184 self.assertEqual(len(rt_value.nested_row_splits), 1)
187 self.assertAllEqual(splits, rt_value.nested_row_splits[0])
196 self.assertEqual(len(rt_value.nested_row_splits), 2)
199 self.assertAllEqual(splits2, rt_value.nested_row_splits[0])
200 self.assertAllEqual(splits, rt_value.nested_row_splits[1])
523 nested_row_splits = [
528 rt = RaggedTensor.from_nested_row_splits(flat_values, nested_row_splits)
539 self.assertIs(rt_row_splits, nested_row_splits[0])
540 self.assertIs(rt_values_row_splits, nested_row_splits[1])
[all …]
Dragged_array_ops.py140 mask.nested_row_splits, data.nested_row_splits[:mask.ragged_rank]
298 nested_splits = rt_input.nested_row_splits
357 nested_splits = rt_input.nested_row_splits
Dragged_math_ops.py279 array_ops.ones_like(data.flat_values), data.nested_row_splits)
293 array_ops.ones_like(data.flat_values), data.nested_row_splits)
536 input_tensor.nested_row_splits)
542 total.flat_values / count.flat_values, total.nested_row_splits)
Dragged_gather_ops.py113 params_nested_splits=params.nested_row_splits,
114 OUTPUT_RAGGED_RANK=indices.shape.ndims + len(params.nested_row_splits) -
Dragged_functional_ops.py102 nested_splits_lists.append(value.nested_row_splits)
Dragged_tensor_value.py70 def nested_row_splits(self): member in RaggedTensorValue
Dragged_concat_ops.py208 nested_splits = [rt.nested_row_splits for rt in rt_inputs]
Dragged_dispatch_test.py148 for (x_splits, y_splits) in zip(x.nested_row_splits, y.nested_row_splits):
Dragged_map_ops.py378 splits = rt.nested_row_splits
Dragged_tensor_shape.py560 src_row_splits = rt_input.nested_row_splits
Dragged_dispatch.py130 elt.nested_row_splits for elt in x if ragged_tensor.is_ragged(elt)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_RaggedGather.pbtxt7 The `nested_row_splits` tensors that define the row-partitioning for the
29 The `nested_row_splits` tensors that define the row-partitioning for the
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.-ragged-tensor.pbtxt15 name: "nested_row_splits"
52 …argspec: "args=[\'cls\', \'flat_values\', \'nested_row_splits\', \'name\'], varargs=None, keywords…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-ragged-tensor.pbtxt15 name: "nested_row_splits"
52 …argspec: "args=[\'cls\', \'flat_values\', \'nested_row_splits\', \'name\'], varargs=None, keywords…
Dtensorflow.ragged.-ragged-tensor-value.pbtxt14 name: "nested_row_splits"