Home
last modified time | relevance | path

Searched refs:is_ragged (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_gather_ops.py131 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) and
404 indices_is_dense = not ragged_tensor.is_ragged(indices)
410 if (indices_is_dense and ragged_tensor.is_ragged(result) and
417 assert not ragged_tensor.is_ragged(indices)
418 assert ragged_tensor.is_ragged(params)
453 if not ragged_tensor.is_ragged(flattened_params):
Dragged_tensor_shape.py185 if not ragged_tensor.is_ragged(rt_input):
205 def is_ragged(self, axis): member in RaggedTensorDynamicShape
331 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):
Dragged_math_ops.py197 if not (ragged_tensor.is_ragged(data) or
198 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):
Dragged_array_ops.py110 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):
Dragged_dispatch.py131 if ragged_tensor.is_ragged(elt):
138 if ragged_tensor.is_ragged(elt) else elt for elt in x
141 ragged_elts = [elt for elt in x if ragged_tensor.is_ragged(elt)]
144 elt.flat_values if ragged_tensor.is_ragged(elt) else elt
154 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 x
226 y_values = y.flat_values if ragged_tensor.is_ragged(y) else y
228 if ragged_tensor.is_ragged(x):
[all …]
Dragged_dispatch_test.py49 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 x
157 if ragged_tensor.is_ragged(result):
270 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x
271 dense_y = y.flat_values if ragged_tensor.is_ragged(y) else y
279 if ragged_tensor.is_ragged(result):
332 x.flat_values if ragged_tensor.is_ragged(x) else x for x in inputs
341 if ragged_tensor.is_ragged(result):
Dragged_concat_ops.py175 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):
Dragged_conversion_ops.py36 if ragged_tensor.is_ragged(tensor):
49 if ragged_tensor.is_ragged(rt_input):
Dragged_string_ops.py136 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):
Dragged_functional_ops.py160 if ragged_tensor.is_ragged(value):
Dragged_batch_gather_with_default_op.py107 if ragged_tensor.is_ragged(params):
Dragged_factory_ops.py171 if ragged_tensor.is_ragged(pylist):
Dragged_cross_op_test.py388 if ragged_tensor.is_ragged(t):
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer.py55 if ragged_tensor.is_ragged(input_tensor):
98 if ragged_tensor.is_ragged(input_tensor):
/external/tensorflow/tensorflow/python/keras/utils/
Dtf_utils_test.py200 self.assertTrue(tf_utils.is_ragged(tensor))
204 self.assertFalse(tf_utils.is_ragged(tensor))
Dtf_utils.py386 def is_ragged(tensor): function
/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Ddiscretization.py238 if tf_utils.is_ragged(inputs):
295 if tf_utils.is_ragged(values):
Dtext_vectorization.py502 if tf_utils.is_ragged(inputs):
566 if tf_utils.is_ragged(lookup_data):
Dcategory_crossing.py154 if any(tf_utils.is_ragged(inp) for inp in inputs):
Dtable_utils.py182 if tf_utils.is_ragged(inputs):
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2235 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 0
3157 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 0
3166 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 0
3167 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 0
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_tensor.py1169 elif ragged_tensor.is_ragged(value):
1483 is_ragged = row_partition.uniform_row_length() is None
1484 if isinstance(value, ops.Tensor) and not is_ragged:
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dtest_base.py249 if sparse_tensor.is_sparse(op1[i]) or ragged_tensor.is_ragged(op1[i]):
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_preprocessing_layer.py402 if tf_utils.is_ragged(values):
Dbase_layer_utils.py233 if tf_utils.is_ragged(tensor):

12