Home
last modified time | relevance | path

Searched refs:TensorArray (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorArrayGradV3.pbtxt9 The handle to the forward TensorArray.
21 The gradient source string, used to decide which gradient TensorArray
25 summary: "Creates a TensorArray for storing the gradients of values in the given handle."
27 If the given TensorArray gradient already exists, returns a reference to it.
29 Locks the size of the original TensorArray by disabling its dynamic size flag.
35 the forward TensorArray is dynamically sized, writes to this TensorArray
36 may resize the object. The gradient TensorArray is statically sized based
37 on the size of the forward TensorArray when this operation executes.
38 Furthermore, the size of the forward TensorArray is frozen by this call.
40 TensorArray only happens after all writes are executed.
[all …]
Dapi_def_TensorArrayGradWithShape.pbtxt9 The handle to the forward TensorArray.
23 elements in the TensorArray corresponding to the input handle.
29 The gradient source string, used to decide which gradient TensorArray
33 summary: "Creates a TensorArray for storing multiple gradients of values in the given handle."
36 expanded shape compared to the input TensorArray whose gradient is being
37 computed. This enables multiple gradients for the same TensorArray to be
Dapi_def_TensorArraySplitV3.pbtxt9 The handle to a TensorArray.
15 The concatenated tensor to write to the TensorArray.
22 TensorArray.
37 summary: "Split the data from the input value into TensorArray elements."
47 this splits values into a TensorArray with T tensors.
49 TensorArray index t will be the subtensor of values with starting position
Dapi_def_TensorArrayV3.pbtxt4 name: "TensorArray"
15 The handle to the TensorArray.
34 validate the shapes of TensorArray elements. If this shape is not
41 A boolean that determines whether writes to the TensorArray
48 If true (default), Tensors in the TensorArray are cleared
57 elements in the TensorArray will be expected to have identical shapes.
68 resource. Default value is the name of the 'TensorArray' op (which
Dapi_def_TensorArraySizeV3.pbtxt9 The handle to a TensorArray (output of TensorArray or TensorArrayGrad).
21 The current size of the TensorArray.
24 summary: "Get the current size of the TensorArray."
Dapi_def_TensorArrayGatherV3.pbtxt9 The handle to a TensorArray.
15 The locations in the TensorArray from which to read tensor elements.
27 All of the elements in the TensorArray, concatenated along a new
41 validate the shapes of TensorArray elements. If this shape is not
45 summary: "Gather specific elements from the TensorArray into output `value`."
Dapi_def_TensorArrayCloseV3.pbtxt9 The handle to a TensorArray (output of TensorArray or TensorArrayGrad).
12 summary: "Delete the TensorArray from its resource container."
Dapi_def_TensorArrayReadV3.pbtxt9 The handle to a TensorArray.
21 The tensor that is read from the TensorArray.
30 summary: "Read an element from the TensorArray into output `value`."
Dapi_def_TensorArrayWriteV3.pbtxt9 The handle to a TensorArray.
15 The position to write to inside the TensorArray.
21 The tensor to write to the TensorArray.
Dapi_def_TensorArrayConcatV3.pbtxt9 The handle to a TensorArray.
21 All of the elements in the TensorArray, concatenated along the first
44 TensorArray elements. If this shape is not fully specified, concatenating
48 summary: "Concat the elements from the TensorArray into value `value`."
Dapi_def_TensorArrayScatterV3.pbtxt9 The handle to a TensorArray.
21 The concatenated tensor to write to the TensorArray.
36 summary: "Scatter the data from the input value into specific TensorArray elements."
/external/tensorflow/tensorflow/compiler/tests/
Dtensor_array_ops_test.py60 ta = tensor_array_ops.TensorArray(
84 ta = tensor_array_ops.TensorArray(
105 ta = tensor_array_ops.TensorArray(
122 ta = tensor_array_ops.TensorArray(
146 ta = tensor_array_ops.TensorArray(
163 ta = tensor_array_ops.TensorArray(
183 ta = tensor_array_ops.TensorArray(
210 ta = tensor_array_ops.TensorArray(
229 ta = tensor_array_ops.TensorArray(
246 ta = tensor_array_ops.TensorArray(
[all …]
Dcond_test.py49 ta = tensor_array_ops.TensorArray(dtype=dtypes.float32, size=1)
69 ta = tensor_array_ops.TensorArray(dtype=dtypes.float32, size=1)
89 ta = tensor_array_ops.TensorArray(dtype=dtypes.float32, size=1)
229 ta = tensor_array_ops.TensorArray(dtype=dtypes.float32, size=1)
251 ta = tensor_array_ops.TensorArray(dtype=dtypes.float32, size=1)
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py66 return tensor_array_ops.TensorArray(
87 ta = tensor_array_ops.TensorArray(
108 ta = tensor_array_ops.TensorArray(
137 ta = tensor_array_ops.TensorArray(
154 ta = tensor_array_ops.TensorArray(
163 ta = tensor_array_ops.TensorArray(
179 tensor_array_ops.TensorArray(
193 ta = tensor_array_ops.TensorArray(
221 ta = tensor_array_ops.TensorArray(
237 ta = tensor_array_ops.TensorArray(
[all …]
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Dknown_issues.md6 ## TensorArray TF/XLA interconversion is not supported
11 XLA supports `tf.TensorArray`. However, the _interconversion_ between TF and
13 This error often arises when the `TensorArray` is used inside the compiled
21 number of elements. This could also happen if you're using a TensorArray in a
27 intermediate results in a `TensorArray`, but XLA only supports bounded
28 `TensorArray`s.
34 ## Dynamic `tf.TensorArray` is not supported
36 Writes into `tf.TensorArray(..., dynamic_size=True)` are not compilable with
/external/tensorflow/tensorflow/python/ops/
Dtensor_array_grad.py108 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
140 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
171 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
201 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
232 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
262 g = (tensor_array_ops.TensorArray(dtype=dtype, handle=handle, flow=flow,
Dtensor_array_ops_test.py36 values = tensor_array_ops.TensorArray(
54 values = tensor_array_ops.TensorArray(
67 values = tensor_array_ops.TensorArray(
80 values = tensor_array_ops.TensorArray(
Dtensor_array_ops.py242 g = TensorArray(
947 class TensorArray(object): class
1275 impl = (old_ta._implementation if isinstance(old_ta, TensorArray)
1289 new_ta = TensorArray(
1322 value_type = property(lambda self: TensorArray)
1369 if not isinstance(value, TensorArray):
1386 ret = TensorArray(
1396 if not isinstance(value, TensorArray):
1415 return TensorArray
1421 TensorArray, TensorArraySpec.from_value, allow_subclass=True)
/external/tensorflow/tensorflow/python/data/util/
Dstructure_test.py60 ("TensorArray", lambda: tensor_array_ops.TensorArray(
108 ("TensorArray", lambda: tensor_array_ops.TensorArray(
110 tensor_array_ops.TensorArray(
112 tensor_array_ops.TensorArray(
115 tensor_array_ops.TensorArray(
117 tensor_array_ops.TensorArray(
191 lambda: tensor_array_ops.TensorArray(
193 lambda: tensor_array_ops.TensorArray(
195 lambda: tensor_array_ops.TensorArray(
263 ("TensorArray", lambda: tensor_array_ops.TensorArray(
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array.cc82 std::atomic<int64> TensorArray::tensor_array_counter{0};
84 Status TensorArray::CopyShapesFrom(TensorArray* rhs, in CopyShapesFrom()
Dtensor_array_ops.cc76 Status GetTensorArray(OpKernelContext* ctx, TensorArray** tensor_array) { in GetTensorArray()
121 TensorArray* output_tensor_array; in Compute()
151 TensorArray** output_tensor_array) = 0;
184 TensorArray** output_tensor_array) override { in CreateTensorArray()
201 TensorArray::tensor_array_counter.fetch_add(1)); in CreateTensorArray()
207 TensorArray* tensor_array = new TensorArray( in CreateTensorArray()
281 TensorArray** output_tensor_array) override { in CreateTensorArray()
307 TensorArray* tensor_array; in CreateTensorArray()
354 tensor_array_output_handle](TensorArray** ret) -> Status { in CreateTensorArray()
355 *ret = new TensorArray( in CreateTensorArray()
[all …]
Dtensor_array.h130 class TensorArray : public ResourceBase {
138 TensorArray(const string& key, const DataType& dtype, const Tensor& handle, in TensorArray() function
335 Status CopyShapesFrom(TensorArray* rhs, const TensorShape* shape_to_prepend);
348 return ctx->step_container()->MakeResourceHandle<TensorArray>( in resource_handle()
442 Status TensorArray::LockedWriteOrAggregate(OpKernelContext* ctx, in LockedWriteOrAggregate()
546 Status TensorArray::LockedRead(OpKernelContext* ctx, const int32 index, in LockedRead()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_side_effects.h34 struct TensorArray : ::mlir::SideEffects::Resource::Base<TensorArray> { struct
/external/tensorflow/tensorflow/python/autograph/operators/
Ddata_structures.py96 l = tensor_array_ops.TensorArray(
189 if isinstance(list_, tensor_array_ops.TensorArray):
259 if isinstance(list_, tensor_array_ops.TensorArray):
323 if isinstance(list_, tensor_array_ops.TensorArray):
Dslices.py57 if isinstance(target, tensor_array_ops.TensorArray):
117 if isinstance(target, tensor_array_ops.TensorArray):

1234