Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dconvert_to_tensor_or_ragged_tensor_op_test.py53 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
82 ragged_tensor.convert_to_tensor_or_ragged_tensor(rt, dtype,
120 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
139 ragged_tensor.convert_to_tensor_or_ragged_tensor(value, dtype,
152 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
175 ragged_tensor.convert_to_tensor_or_ragged_tensor(tensor, dtype,
208 converted = ragged_tensor.convert_to_tensor_or_ragged_tensor(
225 ragged_tensor.convert_to_tensor_or_ragged_tensor(value, dtype,
H A Dragged_where_op.py84 condition = ragged_tensor.convert_to_tensor_or_ragged_tensor(
89 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
90 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, name='y')
169 condition = ragged_tensor.convert_to_tensor_or_ragged_tensor(
174 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
175 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, name='y')
H A Dragged_array_ops.py101 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
102 mask = ragged_tensor.convert_to_tensor_or_ragged_tensor(
240 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
440 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
563 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
614 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
618 partitions = ragged_tensor.convert_to_tensor_or_ragged_tensor(
711 tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(
824 t = ragged_tensor.convert_to_tensor_or_ragged_tensor(t)
975 value = ragged_tensor.convert_to_tensor_or_ragged_tensor(
[all …]
H A Dragged_string_ops.py69 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input,
132 input_tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(input)
414 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input, name="input")
509 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
646 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
768 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(
893 rt = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt)
H A Dragged_math_ops.py210 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
211 segment_ids = ragged_tensor.convert_to_tensor_or_ragged_tensor(
574 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(
702 input_tensor = ragged_tensor.convert_to_tensor_or_ragged_tensor(
845 a = ragged_tensor.convert_to_tensor_or_ragged_tensor(a, name='a')
846 b = ragged_tensor.convert_to_tensor_or_ragged_tensor(b, name='b')
1101 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
1117 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
1133 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, name='x')
H A Dragged_batch_gather_with_default_op.py66 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
69 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
72 default_value = ragged_tensor.convert_to_tensor_or_ragged_tensor(
H A Dragged_dispatch.py31 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x)
42 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(
44 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(
H A Dragged_gather_ops.py83 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
85 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
391 params = ragged_tensor.convert_to_tensor_or_ragged_tensor(
393 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
H A Dragged_dispatch_test.py223 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x)
650 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
651 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
683 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
684 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
H A Dragged_tensor_shape_test.py472 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
473 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
H A Dragged_tensor_shape.py180 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt_input)
494 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt_input)
H A Dragged_functional_ops.py160 value = ragged_tensor.convert_to_tensor_or_ragged_tensor(value)
H A Dragged_squeeze_op.py52 input = ragged_tensor.convert_to_tensor_or_ragged_tensor(input)
H A Ddynamic_ragged_shape.py1640 rt_input = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt_input)
1733 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(
1735 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(
1799 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(
1801 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(
H A Dragged_concat_ops.py146 ragged_tensor.convert_to_tensor_or_ragged_tensor(
H A Dragged_cross_op_test.py417 return ragged_tensor.convert_to_tensor_or_ragged_tensor(t).to_sparse()
H A Dragged_tensor.py2657 def convert_to_tensor_or_ragged_tensor(value, function
2718 return convert_to_tensor_or_ragged_tensor(value, name="values")
H A Ddynamic_ragged_shape_test.py1678 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
1679 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
2206 x = ragged_tensor.convert_to_tensor_or_ragged_tensor(x, dtype=dtypes.int32)
2207 y = ragged_tensor.convert_to_tensor_or_ragged_tensor(y, dtype=dtypes.int32)
H A Dragged_tensor_test.py1568 g = ragged_tensor.convert_to_tensor_or_ragged_tensor(g)
1939 rt = ragged_tensor.convert_to_tensor_or_ragged_tensor(rt)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dbincount_ops.py141 arr = ragged_tensor.convert_to_tensor_or_ragged_tensor(arr, name="arr")
144 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(
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/python/
H A 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)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/util/
H A Dstructure.py115 ragged_tensor.convert_to_tensor_or_ragged_tensor(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/
H A Dstructured_array_ops.py144 indices = ragged_tensor.convert_to_tensor_or_ragged_tensor(
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dmetrics_nonportable_test.py345 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/metrics/
H A Dmetrics_nonportable_test.py345 data = ragged_tensor.convert_to_tensor_or_ragged_tensor(data, name='data')

12