Searched refs:is_ragged (Results 1 – 25 of 29) sorted by relevance
12
131 params_is_ragged = ragged_tensor.is_ragged(params)132 indices_is_ragged = ragged_tensor.is_ragged(indices)374 if not (ragged_tensor.is_ragged(params) or ragged_tensor.is_ragged(indices)):390 if (ragged_tensor.is_ragged(indices) and404 indices_is_dense = not ragged_tensor.is_ragged(indices)410 if (indices_is_dense and ragged_tensor.is_ragged(result) and417 assert not ragged_tensor.is_ragged(indices)418 assert ragged_tensor.is_ragged(params)453 if not ragged_tensor.is_ragged(flattened_params):
185 if not ragged_tensor.is_ragged(rt_input):205 def is_ragged(self, axis): member in RaggedTensorDynamicShape331 if self.is_ragged(axis):356 if self.is_ragged(axis):378 elif self.is_ragged(axis):547 if ragged_tensor.is_ragged(rt_input):556 if ragged_tensor.is_ragged(rt_input):573 if not src_shape.is_ragged(axis) and not dst_shape.is_ragged(axis):595 if not src_shape.is_ragged(axis) and dst_shape.is_ragged(axis):607 if not ragged_tensor.is_ragged(rt_input):
197 if not (ragged_tensor.is_ragged(data) or198 ragged_tensor.is_ragged(segment_ids)):215 if ragged_tensor.is_ragged(segment_ids):216 if not ragged_tensor.is_ragged(data):314 if ragged_tensor.is_ragged(total):330 if ragged_tensor.is_ragged(total):479 if not ragged_tensor.is_ragged(rt_input):616 if ragged_tensor.is_ragged(input_tensor):624 if ragged_tensor.is_ragged(total):
110 if ragged_tensor.is_ragged(mask):111 if not ragged_tensor.is_ragged(data):127 while ragged_tensor.is_ragged(mask):151 elif ragged_tensor.is_ragged(data) and mask.shape.ndims == 1:172 if ragged_tensor.is_ragged(data):237 if not ragged_tensor.is_ragged(input):437 if not ragged_tensor.is_ragged(input):450 if ragged_tensor.is_ragged(input.values):479 if ragged_tensor.is_ragged(input):508 if not ragged_tensor.is_ragged(input):
131 if ragged_tensor.is_ragged(elt):138 if ragged_tensor.is_ragged(elt) else elt for elt in x141 ragged_elts = [elt for elt in x if ragged_tensor.is_ragged(elt)]144 elt.flat_values if ragged_tensor.is_ragged(elt) else elt154 found_ragged = ragged_tensor.is_ragged(x)197 x_is_ragged = ragged_tensor.is_ragged(x)198 y_is_ragged = ragged_tensor.is_ragged(y)225 x_values = x.flat_values if ragged_tensor.is_ragged(x) else x226 y_values = y.flat_values if ragged_tensor.is_ragged(y) else y228 if ragged_tensor.is_ragged(x):[all …]
49 if ragged_tensor.is_ragged(x):50 self.assertTrue(ragged_tensor.is_ragged(y))150 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x157 if ragged_tensor.is_ragged(result):270 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x271 dense_y = y.flat_values if ragged_tensor.is_ragged(y) else y279 if ragged_tensor.is_ragged(result):332 x.flat_values if ragged_tensor.is_ragged(x) else x for x in inputs341 if ragged_tensor.is_ragged(result):
175 if all(not ragged_tensor.is_ragged(rt) for rt in rt_inputs):185 if not ragged_tensor.is_ragged(rt_inputs[i]):303 if not ragged_tensor.is_ragged(rt_input):
36 if ragged_tensor.is_ragged(tensor):49 if ragged_tensor.is_ragged(rt_input):
136 if ragged_tensor.is_ragged(input_tensor):422 if not ragged_tensor.is_ragged(input):432 if ragged_tensor.is_ragged(input):828 if tensor_util.is_tf_type(inputs) or ragged_tensor.is_ragged(inputs):839 if ragged_tensor.is_ragged(input):
160 if ragged_tensor.is_ragged(value):
107 if ragged_tensor.is_ragged(params):
171 if ragged_tensor.is_ragged(pylist):
388 if ragged_tensor.is_ragged(t):
55 if ragged_tensor.is_ragged(input_tensor):98 if ragged_tensor.is_ragged(input_tensor):
200 self.assertTrue(tf_utils.is_ragged(tensor))204 self.assertFalse(tf_utils.is_ragged(tensor))
386 def is_ragged(tensor): function
238 if tf_utils.is_ragged(inputs):295 if tf_utils.is_ragged(values):
502 if tf_utils.is_ragged(inputs):566 if tf_utils.is_ragged(lookup_data):
154 if any(tf_utils.is_ragged(inp) for inp in inputs):
182 if tf_utils.is_ragged(inputs):
2235 elif ragged_tensor.is_ragged(tensor):2721 if ragged_tensor.is_ragged(a) or ragged_tensor.is_ragged(b):2807 if (ragged_tensor.is_ragged(a) or ragged_tensor.is_ragged(b)):3156 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 03157 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 03166 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 03167 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 0
1169 elif ragged_tensor.is_ragged(value):1483 is_ragged = row_partition.uniform_row_length() is None1484 if isinstance(value, ops.Tensor) and not is_ragged:
249 if sparse_tensor.is_sparse(op1[i]) or ragged_tensor.is_ragged(op1[i]):
402 if tf_utils.is_ragged(values):
233 if tf_utils.is_ragged(tensor):