Searched refs:tensor_array (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | tensor_array_ops.cc | 75 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 …]
|
| D | tensor_array.h | 40 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()
|
| D | tensor_array.cc | 28 namespace tensor_array { namespace
|
| D | BUILD | 2012 ":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/ |
| D | subscribe_test.py | 260 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/ |
| D | api_def_TensorArrayV3.pbtxt | 27 The type of the elements on the tensor_array.
67 Overrides the name used for the temporary tensor_array
|
| D | api_def_TensorArrayWriteV3.pbtxt | 36 summary: "Push an element onto the tensor_array."
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | model.py | 805 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/ |
| D | odes.py | 243 def _ta_append(tensor_array, value): argument
245 return tensor_array.write(tensor_array.size(), value)
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass_test.cc | 893 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/ |
| D | tf_op_files.txt | 272 tensorflow/core/kernels/tensor_array.cc
|