Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dtensor_array_ops.cc75 Status GetTensorArray(OpKernelContext* ctx, TensorArray** tensor_array) { in GetTensorArray() argument
83 container + ta_handle, tensor_array)); in GetTensorArray()
86 return LookupResource(ctx, HandleFromInput(ctx, 0), tensor_array); in GetTensorArray()
206 TensorArray* tensor_array = new TensorArray( in CreateTensorArray() local
213 rm->Create(ctx->step_container()->name(), key, tensor_array)); in CreateTensorArray()
215 *output_tensor_array = tensor_array; in CreateTensorArray()
308 TensorArray* tensor_array; in CreateTensorArray() local
311 &tensor_array)); in CreateTensorArray()
312 core::ScopedUnref unref(tensor_array); in CreateTensorArray()
316 tensor_array->DisableDynamicSize(); in CreateTensorArray()
[all …]
Dtensor_array.h40 namespace tensor_array {
520 Status s = tensor_array::AddToTensor<Device, T>(ctx, existing_t, in LockedWriteOrAggregate()
528 Status s = tensor_array::AddToTensor<Device, T>(ctx, local_tensor_t, in LockedWriteOrAggregate()
615 Status s = tensor_array::TensorSetZero<Device, T>(ctx, tensor_t); in LockedRead()
Dtensor_array.cc28 namespace tensor_array { namespace
DBUILD2012 ":tensor_array",
2343 name = "tensor_array",
2344 srcs = ["tensor_array.cc"],
2345 hdrs = ["tensor_array.h"],
5636 "tensor_array.h",
5823 "tensor_array.cc",
/external/tensorflow/tensorflow/python/framework/
Dsubscribe_test.py260 tensor_array = tensor_array_ops.TensorArray(
265 writer = tensor_array.write(0, [[4.0, 5.0]])
278 tensor_array.handle, lambda t: script_ops.py_func(sub, [t], [t.dtype]))
279 self.assertIs(tensor_array_sub, tensor_array.handle)
280 self.assertFalse(subscribe._is_subscribed_identity(tensor_array.handle))
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TensorArrayV3.pbtxt27 The type of the elements on the tensor_array.
67 Overrides the name used for the temporary tensor_array
Dapi_def_TensorArrayWriteV3.pbtxt36 summary: "Push an element onto the tensor_array."
/external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/
Dmodel.py805 def _stack_and_transpose(tensor_array): argument
808 return array_ops.stack(tensor_array.values, axis=1)
813 stacked = tensor_array.stack()
/external/tensorflow/tensorflow/contrib/integrate/python/ops/
Dodes.py243 def _ta_append(tensor_array, value): argument
245 return tensor_array.write(tensor_array.size(), value)
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass_test.cc893 ops::TensorArray tensor_array(root.WithOpName("test/tensor_array"), 1, in TEST() local
897 root.WithOpName("test/write"), tensor_array.handle, zero, in TEST()
898 ops::Const(root.WithOpName("test/forty_two"), 42.0f), tensor_array.flow); in TEST()
900 ops::TensorArrayRead(root.WithOpName("test/read"), tensor_array.handle, in TEST()
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_op_files.txt272 tensorflow/core/kernels/tensor_array.cc