Searched refs:from_row_splits (Results 1 – 17 of 17) sorted by relevance
39 rt1 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 2, 5, 6, 6, 7])40 rt2 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 7])41 rt3 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 0, 7, 7])48 rt1 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 2, 5, 6, 6, 7])49 rt2 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 7])50 rt3 = ragged_tensor.RaggedTensor.from_row_splits(values, [0, 0, 7, 7])56 rt = ragged_tensor.RaggedTensor.from_row_splits(b'a b c d e f g'.split(),
147 bad_rt1 = ragged_tensor.RaggedTensor.from_row_splits(153 bad_rt2 = ragged_tensor.RaggedTensor.from_row_splits(155 bad_rt3 = ragged_tensor.RaggedTensor.from_row_splits(157 values=ragged_tensor.RaggedTensor.from_row_splits(165 bad_rt4 = ragged_tensor.RaggedTensor.from_row_splits(167 values=ragged_tensor.RaggedTensor.from_row_splits(174 bad_rt5 = ragged_tensor.RaggedTensor.from_row_splits(
126 rt = RaggedTensor.from_row_splits(133 rt1 = RaggedTensor.from_row_splits(values, row_splits=[0, 4, 4, 7, 8, 8])144 inner_rt = RaggedTensor.from_row_splits(146 outer_rt = RaggedTensor.from_row_splits(163 rt = RaggedTensor.from_row_splits(360 rt = RaggedTensor.from_row_splits(values, row_splits)379 RaggedTensor.from_row_splits([], [])595 RaggedTensor.from_row_splits, values, splits)605 rt1 = RaggedTensor.from_row_splits(values, row_splits)628 rt1 = RaggedTensor.from_row_splits(values, row_splits)[all …]
186 return ragged_tensor.RaggedTensor.from_row_splits(266 return ragged_tensor.RaggedTensor.from_row_splits(permuted_rt,273 return ragged_tensor.RaggedTensor.from_row_splits(permuted_rt.values,
117 ragged_input_tensor = ragged_tensor.RaggedTensor.from_row_splits(388 codepoints = ragged_tensor.RaggedTensor.from_row_splits(393 offsets = ragged_tensor.RaggedTensor.from_row_splits(
90 return ragged_tensor.RaggedTensor.from_row_splits(109 return ragged_tensor.RaggedTensor.from_row_splits(
140 return ragged_tensor.RaggedTensor.from_row_splits(208 return ragged_tensor.RaggedTensor.from_row_splits(249 return ragged_tensor.RaggedTensor.from_row_splits(inner_rt,
189 return ragged_tensor.RaggedTensor.from_row_splits(masked_values,221 masked_values = ragged_tensor.RaggedTensor.from_row_splits(484 return ragged_tensor.RaggedTensor.from_row_splits(values, splits)
199 x = ragged_tensor.RaggedTensor.from_row_splits([3, 1, 4, 1, 5], splits1)200 y = ragged_tensor.RaggedTensor.from_row_splits([1, 2, 3, 4, 5], splits2)
365 def from_row_splits(cls, values, row_splits, name=None): member in RaggedTensor589 result = cls.from_row_splits(result, splits)1210 return cls.from_row_splits(values, splits)1220 return cls.from_row_splits(values, splits)
103 return ragged_tensor.RaggedTensor.from_row_splits(result.rt_dense_values,241 return ragged_tensor.RaggedTensor.from_row_splits(output_values,
74 return _constant_value(ragged_tensor.RaggedTensor.from_row_splits,
211 segment_ids2 = ragged_tensor.RaggedTensor.from_row_splits(
477 ragged_indices = ragged_tensor.RaggedTensor.from_row_splits(
445 y = ragged_tensor.RaggedTensor.from_row_splits(
67 name: "from_row_splits"