Home
last modified time | relevance | path

Searched refs:tensor_array_name (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dtensor_array_ops_test.py57 dtype=dtypes.float32, tensor_array_name="foo", size=3)
81 dtype=tf_dtype, tensor_array_name="foo", size=3)
102 dtype=dtypes.float32, tensor_array_name="foo", size=3)
119 dtype=tf_dtype, tensor_array_name="foo", size=3)
143 dtype=tf_dtype, tensor_array_name="foo", size=3)
160 dtype=tf_dtype, tensor_array_name="foo", size=3)
180 dtype=tf_dtype, tensor_array_name="foo", size=3)
207 dtype=tf_dtype, tensor_array_name="foo", size=3)
226 dtype=tf_dtype, tensor_array_name="foo", size=3)
243 dtype=tf_dtype, tensor_array_name="foo", size=3)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dtensor_array_ops_test.py63 dtype=dtype, tensor_array_name=name, size=size, infer_shape=infer_shape)
85 tensor_array_name="foo",
105 dtype=tf_dtype, tensor_array_name="foo", size=3)
134 dtype=dtypes.float32, tensor_array_name="foo", size=3)
190 dtype=tf_dtype, tensor_array_name="foo", size=3, infer_shape=False)
219 tensor_array_name="foo",
235 tensor_array_name="foo",
253 tensor_array_name="foo",
367 tensor_array_name="foo",
405 tensor_array_name="foo",
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtensor_array_ops_decomposition.mlir12 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
45 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
63 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
79 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
93 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
109 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
133 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
155 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
201 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
248 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dtensor_array_ops.py56 tensor_array_name=None, argument
97 if handle is not None and tensor_array_name:
164 tensor_array_name=tensor_array_name,
405 tensor_array_name=None, argument
439 del tensor_array_name
668 tensor_array_name=None, argument
698 del (flow, tensor_array_name, name) # Unused.
1024 tensor_array_name=None, argument
1090 tensor_array_name=tensor_array_name,
H A Drnn.py868 tensor_array_name=base_name + name)
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DTensorArrayV2.pbtxt39 name: "tensor_array_name"
85 name: "tensor_array_name"
H A DTensorArrayV3.pbtxt43 name: "tensor_array_name"
100 name: "tensor_array_name"
H A DTensorArray.pbtxt31 name: "tensor_array_name"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DTensorArrayV2.pbtxt39 name: "tensor_array_name"
85 name: "tensor_array_name"
H A DTensorArrayV3.pbtxt43 name: "tensor_array_name"
100 name: "tensor_array_name"
H A DTensorArray.pbtxt31 name: "tensor_array_name"
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/tests/
H A Dtensor_array_size.mlir9 …lement_shape = #tf_type.shape<*>, identical_element_shapes = false, tensor_array_name = ""} : (ten…
11 …lement_shape = #tf_type.shape<*>, identical_element_shapes = false, tensor_array_name = ""} : (ten…
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_array_ops.cc283 string tensor_array_name; in CreateTensorArray() local
285 TF_RETURN_IF_ERROR(GetHandle(ctx, &container, &tensor_array_name)); in CreateTensorArray()
299 tensor_array_name = in CreateTensorArray()
305 output_handle(1) = strings::StrCat(tensor_array_name, "@", source_); in CreateTensorArray()
309 rm, strings::StrCat(container, tensor_array_name), &tensor_array)); in CreateTensorArray()
326 "Unable to create a gradients TensorArray for ", tensor_array_name, in CreateTensorArray()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dfunc_use_fallback_tensor.mlir145 …ent_shape = #tf_type.shape<?x100>, identical_element_shapes = true, tensor_array_name = "processed…
146 …ent_shape = #tf_type.shape<?x512>, identical_element_shapes = true, tensor_array_name = "processed…
154 …ent_shape = #tf_type.shape<?x100>, identical_element_shapes = true, tensor_array_name = "processed…
155 …ent_shape = #tf_type.shape<?x512>, identical_element_shapes = true, tensor_array_name = "processed…
H A Dattributes.mlir36 … = false, clear_after_read = true, identical_element_shapes = true, tensor_array_name = "ta"} : (t…
H A Dfallback.mlir115 …ent_shape = #tf_type.shape<?x512>, identical_element_shapes = true, tensor_array_name = "output"} …
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/
H A Dgarbage_collection_test.py40 tensor_array_name="foo",
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.-tensor-array.pbtxt27 …f\', \'dtype\', \'size\', \'dynamic_size\', \'clear_after_read\', \'tensor_array_name\', \'handle\…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.-tensor-array.pbtxt27 …f\', \'dtype\', \'size\', \'dynamic_size\', \'clear_after_read\', \'tensor_array_name\', \'handle\…
H A Dtensorflow.raw_ops.pbtxt4937 …rgs=[\'size\', \'dtype\', \'dynamic_size\', \'clear_after_read\', \'tensor_array_name\', \'element…
5049 …ype\', \'element_shape\', \'dynamic_size\', \'clear_after_read\', \'tensor_array_name\', \'name\']…
5053 …namic_size\', \'clear_after_read\', \'identical_element_shapes\', \'tensor_array_name\', \'name\']…
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_TensorArrayV3.pbtxt65 name: "tensor_array_name"
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsubscribe_test.py258 tensor_array_name='test',
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/
H A Dbackend.py4375 tensor_array_name='input_ta_%s' % i)
4396 tensor_array_name='output_ta_%s' % i)
4422 tensor_array_name='mask_ta')
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py3594 tensor_array_name = pfor_input.get_attr("tensor_array_name")
3604 tensor_array_name=tensor_array_name)
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v0.pbtxt36816 name: "tensor_array_name"
36854 name: "tensor_array_name"
36892 name: "tensor_array_name"
36933 name: "tensor_array_name"
38554 name: "tensor_array_name"
38600 name: "tensor_array_name"
38646 name: "tensor_array_name"
38695 name: "tensor_array_name"
38745 name: "tensor_array_name"

12