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
|