Home
last modified time | relevance | path

Searched refs:tensor_array (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtensor_array_ops.cc76 Status GetTensorArray(OpKernelContext* ctx, TensorArray** tensor_array) { in GetTensorArray() argument
84 ctx->step_container()->Lookup(rm, container + ta_handle, tensor_array)); in GetTensorArray()
87 return LookupResource(ctx, HandleFromInput(ctx, 0), tensor_array); in GetTensorArray()
207 TensorArray* tensor_array = new TensorArray( in CreateTensorArray() local
213 TF_RETURN_IF_ERROR(ctx->step_container()->Create(rm, key, tensor_array)); in CreateTensorArray()
215 *output_tensor_array = tensor_array; in CreateTensorArray()
307 TensorArray* tensor_array; in CreateTensorArray() local
309 rm, strings::StrCat(container, tensor_array_name), &tensor_array)); in CreateTensorArray()
310 core::ScopedUnref unref(tensor_array); in CreateTensorArray()
314 tensor_array->DisableDynamicSize(); in CreateTensorArray()
[all …]
H A Dtensor_array.h40 namespace tensor_array {
516 Status s = tensor_array::AddToTensor<Device, T>(ctx, existing_t, in LockedWriteOrAggregate()
523 Status s = tensor_array::AddToTensor<Device, T>(ctx, &local_tensor, in LockedWriteOrAggregate()
608 Status s = tensor_array::TensorSetZero<Device, T>(ctx, &t.tensor); in LockedRead()
H A Dtensor_array.cc29 namespace tensor_array { namespace
H A DBUILD2601 ":tensor_array",
2934 name = "tensor_array",
2935 srcs = ["tensor_array.cc"],
2936 hdrs = ["tensor_array.h"],
6512 "tensor_array.h",
6828 "tensor_array.cc",
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dsubscribe_test.py256 tensor_array = tensor_array_ops.TensorArray(
261 writer = tensor_array.write(0, [[4.0, 5.0]])
274 tensor_array.handle, lambda t: script_ops.py_func(sub, [t], [t.dtype]))
275 self.assertIs(tensor_array_sub, tensor_array.handle)
276 self.assertFalse(subscribe._is_subscribed_identity(tensor_array.handle))
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_TensorArrayV3.pbtxt27 The type of the elements on the tensor_array.
67 Overrides the name used for the temporary tensor_array
H A Dapi_def_TensorArrayWriteV3.pbtxt36 summary: "Push an element onto the tensor_array."
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/
H A Dfallback.mlir106 // CHECK-LABEL: func @tensor_array
107 func.func @tensor_array() -> (tensor<1x1x512xf32>) {
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass_test.cc1064 ops::TensorArray tensor_array(root.WithOpName("test/tensor_array"), 1, in TEST() local
1068 root.WithOpName("test/write"), tensor_array.handle, zero, in TEST()
1069 ops::Const(root.WithOpName("test/forty_two"), 42.0f), tensor_array.flow); in TEST()
1071 ops::TensorArrayRead(root.WithOpName("test/read"), tensor_array.handle, in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td18235 let summary = "Push an element onto the tensor_array.";