| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_gather_ops.py | 129 params_is_ragged = ragged_tensor.is_ragged(params)
130 indices_is_ragged = ragged_tensor.is_ragged(indices)
386 if not (ragged_tensor.is_ragged(params) or ragged_tensor.is_ragged(indices)):
402 if (ragged_tensor.is_ragged(indices) and
416 indices_is_dense = not ragged_tensor.is_ragged(indices)
422 if (indices_is_dense and ragged_tensor.is_ragged(result) and
429 assert not ragged_tensor.is_ragged(indices)
430 assert ragged_tensor.is_ragged(params)
465 if not ragged_tensor.is_ragged(flattened_params):
|
| D | ragged_tensor_shape.py | 181 if not ragged_tensor.is_ragged(rt_input):
201 def is_ragged(self, axis): member in RaggedTensorDynamicShape
328 if self.is_ragged(axis):
353 if self.is_ragged(axis):
375 elif self.is_ragged(axis):
546 if ragged_tensor.is_ragged(rt_input):
555 if ragged_tensor.is_ragged(rt_input):
572 if not src_shape.is_ragged(axis) and not dst_shape.is_ragged(axis):
594 if not src_shape.is_ragged(axis) and dst_shape.is_ragged(axis):
606 if not ragged_tensor.is_ragged(rt_input):
|
| D | ragged_dispatch.py | 38 x_is_ragged = ragged_tensor.is_ragged(x)
39 y_is_ragged = ragged_tensor.is_ragged(y)
70 x_values = x.flat_values if ragged_tensor.is_ragged(x) else x
71 y_values = y.flat_values if ragged_tensor.is_ragged(y) else y
75 if ragged_tensor.is_ragged(x):
|
| D | ragged_array_ops.py | 114 if ragged_tensor.is_ragged(mask):
115 if not ragged_tensor.is_ragged(data):
131 while ragged_tensor.is_ragged(mask):
155 elif ragged_tensor.is_ragged(data) and mask.shape.ndims == 1:
176 if ragged_tensor.is_ragged(data):
242 if not ragged_tensor.is_ragged(input):
443 if not ragged_tensor.is_ragged(input):
456 if ragged_tensor.is_ragged(input.values):
497 if ragged_tensor.is_ragged(input):
535 if not ragged_tensor.is_ragged(input):
|
| D | ragged_dispatch_test.py | 56 if ragged_tensor.is_ragged(x):
57 self.assertTrue(ragged_tensor.is_ragged(y))
228 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x
236 if ragged_tensor.is_ragged(result):
374 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x
375 dense_y = y.flat_values if ragged_tensor.is_ragged(y) else y
382 if ragged_tensor.is_ragged(result):
503 dense_x = x.flat_values if ragged_tensor.is_ragged(x) else x
504 dense_y = y.flat_values if ragged_tensor.is_ragged(y) else y
559 x.flat_values if ragged_tensor.is_ragged(x) else x for x in inputs
[all …]
|
| D | ragged_math_ops.py | 199 if not (ragged_tensor.is_ragged(data) or
200 ragged_tensor.is_ragged(segment_ids)):
217 if ragged_tensor.is_ragged(segment_ids):
218 if not ragged_tensor.is_ragged(data):
336 if ragged_tensor.is_ragged(total):
356 if ragged_tensor.is_ragged(total):
521 if not ragged_tensor.is_ragged(rt_input):
678 if ragged_tensor.is_ragged(input_tensor):
686 if ragged_tensor.is_ragged(total):
|
| D | ragged_concat_ops.py | 177 if all(not ragged_tensor.is_ragged(rt) for rt in rt_inputs):
187 if not ragged_tensor.is_ragged(rt_inputs[i]):
305 if not ragged_tensor.is_ragged(rt_input):
|
| D | ragged_string_ops.py | 135 if ragged_tensor.is_ragged(input_tensor):
421 if not ragged_tensor.is_ragged(input):
431 if ragged_tensor.is_ragged(input):
839 if tensor_util.is_tf_type(inputs) or ragged_tensor.is_ragged(inputs):
850 if ragged_tensor.is_ragged(input):
|
| D | ragged_conversion_ops.py | 32 if ragged_tensor.is_ragged(tensor):
45 if ragged_tensor.is_ragged(rt_input):
|
| D | dynamic_ragged_shape.py | 432 if ragged_tensor.is_ragged(t):
1642 if ragged_tensor.is_ragged(rt_input):
1729 x_is_ragged = ragged_tensor.is_ragged(x)
1730 y_is_ragged = ragged_tensor.is_ragged(y)
1764 x_values = x.flat_values if ragged_tensor.is_ragged(x) else x
1765 y_values = y.flat_values if ragged_tensor.is_ragged(y) else y
1769 if ragged_tensor.is_ragged(x):
1795 x_is_ragged = ragged_tensor.is_ragged(x)
1796 y_is_ragged = ragged_tensor.is_ragged(y)
1827 x_values = x.flat_values if ragged_tensor.is_ragged(x) else x
[all …]
|
| D | ragged_functional_ops.py | 159 if ragged_tensor.is_ragged(value):
|
| D | ragged_batch_gather_with_default_op.py | 103 if ragged_tensor.is_ragged(params):
|
| D | ragged_factory_ops.py | 167 if ragged_tensor.is_ragged(pylist):
|
| D | ragged_cross_op_test.py | 416 if ragged_tensor.is_ragged(t):
|
| D | ragged_tensor.py | 2251 def is_ragged(value): function
2498 if is_ragged(value):
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/python/ |
| D | sentencepiece_tokenizer.py | 55 if ragged_tensor.is_ragged(input_tensor):
98 if ragged_tensor.is_ragged(input_tensor):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | test_util.py | 2574 elif ragged_tensor.is_ragged(tensor):
2983 if ragged_tensor.is_ragged(a) or ragged_tensor.is_ragged(b):
3165 if (ragged_tensor.is_ragged(a) or ragged_tensor.is_ragged(b)):
3558 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 0
3559 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 0
3568 a_ragged_rank = a.ragged_rank if ragged_tensor.is_ragged(a) else 0
3569 b_ragged_rank = b.ragged_rank if ragged_tensor.is_ragged(b) else 0
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 1187 elif ragged_tensor.is_ragged(value):
1528 is_ragged = row_partition.uniform_row_length() is None
1529 if isinstance(value, ops.Tensor) and not is_ragged:
|
| D | structured_array_ops.py | 510 if ragged_tensor.is_ragged(t):
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_preprocessing_layer.py | 447 if tf_utils.is_ragged(values):
|
| D | base_layer_utils.py | 233 if tf_utils.is_ragged(tensor):
|
| D | compile_utils.py | 212 if tf_utils.is_ragged(y_t):
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 386 def is_ragged(tensor): function
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | test_base.py | 262 if sparse_tensor.is_sparse(op1[i]) or ragged_tensor.is_ragged(op1[i]):
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_fast_parsing.cc | 632 bool is_ragged = d_and_type.second == Type::Ragged; in FastParseSerializedExample() local
639 : (is_ragged ? config.ragged[d].feature_name in FastParseSerializedExample()
808 is_ragged ? ragged_feature_last_example : sparse_feature_last_example; in FastParseSerializedExample()
819 SparseBuffer& out = is_ragged ? (*output_ragged)[d] : (*output_sparse)[d]; in FastParseSerializedExample()
821 is_ragged ? config.ragged[d].dtype : config.sparse[d].dtype; in FastParseSerializedExample()
|