Searched refs:is_ragged (Results 1 – 14 of 14) sorted by relevance
100 if ragged_tensor.is_ragged(indices):103 if not ragged_tensor.is_ragged(params):168 if not (ragged_tensor.is_ragged(params) or ragged_tensor.is_ragged(indices)):183 if (ragged_tensor.is_ragged(indices) and197 indices_is_dense = not ragged_tensor.is_ragged(indices)202 if (indices_is_dense and ragged_tensor.is_ragged(result) and209 assert not ragged_tensor.is_ragged(indices)210 assert ragged_tensor.is_ragged(params)245 if not ragged_tensor.is_ragged(flattened_params):
55 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 056 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 065 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 066 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 083 if ragged_tensor.is_ragged(value):91 if ragged_tensor.is_ragged(tensor):
133 if ragged_tensor.is_ragged(mask):134 if not ragged_tensor.is_ragged(data):148 while ragged_tensor.is_ragged(mask):173 elif ragged_tensor.is_ragged(data) and mask.shape.ndims == 1:194 if ragged_tensor.is_ragged(data):260 if not ragged_tensor.is_ragged(input):469 if not ragged_tensor.is_ragged(input):511 if ragged_tensor.is_ragged(input):525 if not ragged_tensor.is_ragged(rt_input):575 if not ragged_tensor.is_ragged(input):
67 if not (ragged_tensor.is_ragged(params) or ragged_tensor.is_ragged(indices)):82 if ragged_tensor.is_ragged(indices):85 if not ragged_tensor.is_ragged(params):96 if not ragged_tensor.is_ragged(params):
165 if not ragged_tensor.is_ragged(rt_input):184 def is_ragged(self, axis): member in RaggedTensorDynamicShape304 if self.is_ragged(axis):329 if self.is_ragged(axis):351 elif self.is_ragged(axis):500 if ragged_tensor.is_ragged(rt_input):507 if ragged_tensor.is_ragged(rt_input):522 if not src_shape.is_ragged(axis) and not dst_shape.is_ragged(axis):542 if not src_shape.is_ragged(axis) and dst_shape.is_ragged(axis):553 if not ragged_tensor.is_ragged(rt_input):
184 if not (ragged_tensor.is_ragged(data) or185 ragged_tensor.is_ragged(segment_ids)):194 if ragged_tensor.is_ragged(segment_ids):195 if not ragged_tensor.is_ragged(data):281 if ragged_tensor.is_ragged(total):295 if ragged_tensor.is_ragged(total):440 if not ragged_tensor.is_ragged(rt_input):533 if ragged_tensor.is_ragged(input_tensor):540 if ragged_tensor.is_ragged(total):
124 if ragged_tensor.is_ragged(elt):130 elt.nested_row_splits for elt in x if ragged_tensor.is_ragged(elt)133 elt.flat_values if ragged_tensor.is_ragged(elt) else elt144 found_ragged = ragged_tensor.is_ragged(x)186 x_is_ragged = ragged_tensor.is_ragged(x)187 y_is_ragged = ragged_tensor.is_ragged(y)211 x_values = x.flat_values if ragged_tensor.is_ragged(x) else x212 y_values = y.flat_values if ragged_tensor.is_ragged(y) else y214 if ragged_tensor.is_ragged(x):261 if ragged_tensor.is_ragged(elt):[all …]
25 if ragged_tensor.is_ragged(tensor):33 if ragged_tensor.is_ragged(rt_input):
74 if ragged_tensor.is_ragged(input_tensor):357 if not ragged_tensor.is_ragged(input):367 if ragged_tensor.is_ragged(input):
160 if all(not ragged_tensor.is_ragged(rt) for rt in rt_inputs):170 if not ragged_tensor.is_ragged(rt_inputs[i]):287 if not ragged_tensor.is_ragged(rt_input):
100 if ragged_tensor.is_ragged(value):
110 if ragged_tensor.is_ragged(params):
155 if ragged_tensor.is_ragged(pylist):
1255 if is_ragged(values):1515 def is_ragged(value): function