Home
last modified time | relevance | path

Searched refs:nested_splits (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_variant.h40 RaggedTensorVariant(Tensor values, const std::vector<Tensor>& nested_splits) in RaggedTensorVariant() argument
41 : values_(std::move(values)), nested_splits_(nested_splits) {} in RaggedTensorVariant()
56 const std::vector<Tensor>& nested_splits() const { return nested_splits_; } in nested_splits() function
60 void set_nested_splits(const std::vector<Tensor>& nested_splits) { in set_nested_splits() argument
61 nested_splits_ = nested_splits; in set_nested_splits()
74 y->set_nested_splits(x.nested_splits()); in RaggedTensorVariantZerosLike()
102 out->set_nested_splits(x.nested_splits()); in RaggedTensorVariantBinaryAdd()
Dragged_tensor_from_variant_op.cc63 for (const auto& splits : decoded->nested_splits()) { in RaggedComponentsFromVariant()
119 if (!ragged_components[j].nested_splits().empty()) { in NestedStackRaggedTensors()
131 if (ragged_components[j].nested_splits().empty()) { in NestedStackRaggedTensors()
Dragged_tensor_variant.cc60 *to->mutable_nested_splits() = from.nested_splits(); in RaggedTensorVariantDeviceCopy()
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_factory_ops.py229 nested_splits = []
232 nested_splits.append([0])
235 nested_splits[dim].append(nested_splits[dim][-1] + len(row))
241 for row_splits in reversed(nested_splits):
Dragged_array_ops.py278 nested_splits = rt_input.nested_row_splits
281 inner_value_ids = math_ops.range(nested_splits[-1][-1])
288 splits = nested_splits[axis - 1]
337 nested_splits = rt_input.nested_row_splits
343 projected_splits = [{i: nested_splits[i]} for i in range(ragged_rank)]
347 nested_splits[dst_axis], projected_splits[src_axis][dst_axis - 1])
353 input_lengths = nested_splits[axis][1:] - nested_splits[axis][:-1]
Dragged_functional_ops.py119 … [math_ops.cast(s, dtypes.int64) for s in nested_splits] # pylint: disable=g-complex-comprehension
120 for nested_splits in nested_splits_lists]
Dragged_concat_ops.py224 nested_splits = [rt.nested_row_splits for rt in rt_inputs]
228 for ns in nested_splits])
Dragged_tensor.py1276 nested_splits = self.nested_row_splits
1282 return array_ops.shape(nested_splits[0], out_type=out_type)[0] - 1
1291 for splits in nested_splits
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc157 TfLiteTensor& nested_splits = in Prepare() local
159 SetTensorToDynamic(&nested_splits); in Prepare()