Home
last modified time | relevance | path

Searched refs:convert_to_tensor_or_ragged_tensor (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/ops/ragged/
Dconvert_to_tensor_or_ragged_tensor_op_test.py57 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
86 ragged_tensor.convert_to_tensor_or_ragged_tensor(rt, dtype,
124 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
143 ragged_tensor.convert_to_tensor_or_ragged_tensor(value, dtype,
156 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
179 ragged_tensor.convert_to_tensor_or_ragged_tensor(tensor, dtype,
212 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
229 ragged_tensor.convert_to_tensor_or_ragged_tensor(value, dtype,
Dragged_where_op.py81 condition = ragged_tensor.convert_to_tensor_or_ragged_tensor(
86 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
87 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, name='y')
162 condition = ragged_tensor.convert_to_tensor_or_ragged_tensor(
167 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
168 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, name='y')
Dragged_array_ops.py97 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
98 mask = ragged_tensor.convert_to_tensor_or_ragged_tensor(
235 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
434 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
535 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
586 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
590 partitions = ragged_tensor.convert_to_tensor_or_ragged_tensor(
682 tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(
795 t = ragged_tensor.convert_to_tensor_or_ragged_tensor(t)
Dragged_batch_gather_with_default_op.py70 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
73 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
76 default_value = ragged_tensor.convert_to_tensor_or_ragged_tensor(
Dragged_string_ops.py70 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input,
133 input_tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(input)
415 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input, name="input")
510 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
647 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
764 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(
882 rt = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt)
Dragged_gather_ops.py85 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
87 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
379 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
381 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
Dragged_dispatch.py138 ragged_tensor.convert_to_tensor_or_ragged_tensor(elt)
157 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name=self._x)
205 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(
207 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(
479 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
488 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
Dragged_math_ops.py209 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
210 segment_ids = ragged_tensor.convert_to_tensor_or_ragged_tensor(
549 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
775 a = ragged_tensor.convert_to_tensor_or_ragged_tensor(a, name='a')
776 b = ragged_tensor.convert_to_tensor_or_ragged_tensor(b, name='b')
Dragged_squeeze_op.py53 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input)
Dragged_functional_ops.py161 value = ragged_tensor.convert_to_tensor_or_ragged_tensor(value)
Dragged_tensor_shape_test.py476 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
477 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
Dragged_dispatch_test.py145 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x)
375 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
376 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
Dragged_tensor_shape.py184 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt_input)
495 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt_input)
Dragged_concat_ops.py144 ragged_tensor.convert_to_tensor_or_ragged_tensor(
Dragged_cross_op_test.py389 return ragged_tensor.convert_to_tensor_or_ragged_tensor(t).to_sparse()
Dragged_tensor.py2603 def convert_to_tensor_or_ragged_tensor(value, function
2664 return convert_to_tensor_or_ragged_tensor(value, name="values")
Dragged_tensor_test.py1633 rt = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt)
/external/tensorflow/tensorflow/python/ops/
Dbincount_ops.py144 arr = ragged_tensor.convert_to_tensor_or_ragged_tensor(arr, name="arr")
147 weights = ragged_tensor.convert_to_tensor_or_ragged_tensor(
389 values = ragged_tensor.convert_to_tensor_or_ragged_tensor(
393 weights = ragged_tensor.convert_to_tensor_or_ragged_tensor(
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dsentencepiece_tokenizer.py52 input_tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(inputs)
93 input_tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(input)
/external/tensorflow/tensorflow/python/data/util/
Dstructure.py117 ragged_tensor.convert_to_tensor_or_ragged_tensor(
/external/tensorflow/tensorflow/python/ops/structured/
Dstructured_array_ops.py132 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
Dstructured_tensor.py1264 return ragged_tensor.convert_to_tensor_or_ragged_tensor(value)
/external/tensorflow/tensorflow/lite/python/
Dmetrics_nonportable_test.py345 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')