Searched refs:nested_splits (Results 1 – 9 of 9) sorted by relevance
40 RaggedTensorVariant(Tensor values, const std::vector<Tensor>& nested_splits) in RaggedTensorVariant() argument41 : values_(std::move(values)), nested_splits_(nested_splits) {} in RaggedTensorVariant()56 const std::vector<Tensor>& nested_splits() const { return nested_splits_; } in nested_splits() function60 void set_nested_splits(const std::vector<Tensor>& nested_splits) { in set_nested_splits() argument61 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()
63 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()
60 *to->mutable_nested_splits() = from.nested_splits(); in RaggedTensorVariantDeviceCopy()
229 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):
278 nested_splits = rt_input.nested_row_splits281 inner_value_ids = math_ops.range(nested_splits[-1][-1])288 splits = nested_splits[axis - 1]337 nested_splits = rt_input.nested_row_splits343 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]
119 … [math_ops.cast(s, dtypes.int64) for s in nested_splits] # pylint: disable=g-complex-comprehension120 for nested_splits in nested_splits_lists]
224 nested_splits = [rt.nested_row_splits for rt in rt_inputs]228 for ns in nested_splits])
1276 nested_splits = self.nested_row_splits1282 return array_ops.shape(nested_splits[0], out_type=out_type)[0] - 11291 for splits in nested_splits
157 TfLiteTensor& nested_splits = in Prepare() local159 SetTensorToDynamic(&nested_splits); in Prepare()