Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlist_ops_test.py57 l = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
111 l = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
169 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
197 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
205 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
219 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
228 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
240 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
251 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
261 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dtensor_list_ops_test.py136 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
147 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
157 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
165 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
177 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32))
202 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
224 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.float32)
235 z = list_ops.tensor_list_stack(z, element_dtype=dtypes.float32)
H A Dadd_n_test.py39 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32),
52 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32).eval()
73 list_ops.tensor_list_stack(l, element_dtype=dtypes.float32), {
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/
H A Ddata_structures_test.py41 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
49 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
55 t = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
101 t = list_ops.tensor_list_stack(l, element_dtype=x.dtype)
136 t = list_ops.tensor_list_stack(l, element_dtype=initial_list.dtype)
H A Dslices_test.py32 t = list_ops.tensor_list_stack(l, element_dtype=initial_list.dtype)
H A Ddata_structures.py341 return list_ops.tensor_list_stack(list_, element_dtype=opts.element_dtype)
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/lang/
H A Dspecial_functions_test.py42 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
49 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
56 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
73 sl = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dcontrol_flow_ops_test.py1125 return (list_ops.tensor_list_stack(h1, dtypes.int32),
1126 list_ops.tensor_list_stack(h2, dtypes.int32))
1178 return (list_ops.tensor_list_stack(h1, dtypes.int32),
1179 list_ops.tensor_list_stack(h2, dtypes.int32))
1218 return list_ops.tensor_list_stack(handle, dtypes.int32)
1228 return list_ops.tensor_list_stack(handle, dtypes.int32)
1239 return tensor, list_ops.tensor_list_stack(handle, dtypes.int32)
1250 return tensor, list_ops.tensor_list_stack(handle, dtypes.int32)
1263 stacked = list_ops.tensor_list_stack(handle, dtypes.float32)
1277 return list_ops.tensor_list_stack(handle, dtypes.int32)
[all …]
H A Dpfor.py4137 output = list_ops.tensor_list_stack(
5056 list_ops.tensor_list_stack(
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/converters/
H A Dlists_test.py63 r = list_ops.tensor_list_stack(tl, dtypes.int32)
77 r = list_ops.tensor_list_stack(tl, dtypes.int32)
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dlist_ops.py157 def tensor_list_stack(input_handle, function
162 return gen_list_ops.tensor_list_stack(
276 tensor_grad = gen_list_ops.tensor_list_stack(
H A Dtensor_array_ops.py561 value = list_ops.tensor_list_stack(
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/
H A Dtensor_list_resize.py46 out = list_ops.tensor_list_stack(
H A Dtensor_list_set_item.py47 out = list_ops.tensor_list_stack(
H A Dtensor_list_dynamic_shape.py55 out = list_ops.tensor_list_stack(
H A Ddynamic_update_slice.py65 out = list_ops.tensor_list_stack(
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/mlir/testing/op_tests/
H A Dtensor_list_resize.py49 out = list_ops.tensor_list_stack(
H A Dtensor_list_set_item.py51 out = list_ops.tensor_list_stack(
H A Dtensor_list_dynamic_shape.py59 out = list_ops.tensor_list_stack(
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dwhile_v2_test.py238 return z + math_ops.reduce_sum(list_ops.tensor_list_stack(
289 return z + math_ops.reduce_sum(list_ops.tensor_list_stack(
692 stack = list_ops.tensor_list_stack(outputs[1], element_dtype=x.dtype)
791 stack = list_ops.tensor_list_stack(outputs[1], element_dtype=x.dtype)
1373 list_ops.tensor_list_stack(result, element_dtype=dtypes.float32)),
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/
H A Dtensor_list_test.py40 s = list_ops.tensor_list_stack(l, element_dtype=dtypes.int32)
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Darray_ops_test.py2395 tiled_tensor_0 = list_ops.tensor_list_stack(tiled_handles[0], t.dtype, 2,
2397 tiled_tensor_1 = list_ops.tensor_list_stack(tiled_handles[1], t.dtype, 2,
2406 tiled_tensor_0 = list_ops.tensor_list_stack(tiled_handles[0], t.dtype, 2,
2408 tiled_tensor_1 = list_ops.tensor_list_stack(tiled_handles[1], t.dtype, 2,
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dlite_flex_test.py181 return list_ops.tensor_list_stack(l_final, element_dtype=tf.int64)