MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / TensorArray

Method TensorArray

tensorflow/core/kernels/tensor_array.h:140–157  ·  view source on GitHub ↗

Construct a TensorArray for holding Tensors of type 'dtype' with 'N' elements. While the underlying storage is a std::vector and can hold more than MAX_INT entries, in practice we do not expect users to construct this many Tensors for storage in a TensorArray.

Source from the content-addressed store, hash-verified

138 // can hold more than MAX_INT entries, in practice we do not expect
139 // users to construct this many Tensors for storage in a TensorArray.
140 TensorArray(const string& key, const DataType& dtype, const Tensor& handle,
141 int32 N, const PartialTensorShape& element_shape,
142 bool identical_element_shapes, bool dynamic_size,
143 bool multiple_writes_aggregate, bool is_grad, int32 marked_size,
144 bool clear_after_read)
145 : key_(key),
146 dtype_(dtype),
147 handle_(handle),
148 closed_(false),
149 dynamic_size_(dynamic_size),
150 multiple_writes_aggregate_(multiple_writes_aggregate),
151 gradients_disallowed_(false),
152 clear_after_read_(clear_after_read),
153 is_grad_(is_grad),
154 marked_size_(marked_size),
155 element_shape_(element_shape),
156 identical_element_shapes_(identical_element_shapes),
157 tensors_(N) {}
158
159 // Write PersistentTensor 'value' to index 'index'.
160 //

Callers 15

self_attentionFunction · 0.80
self_all_attentionFunction · 0.80
_create_taFunction · 0.80
raw_rnnFunction · 0.80
self_attentionFunction · 0.80
self_all_attentionFunction · 0.80
_create_taFunction · 0.80
raw_rnnFunction · 0.80
self_attentionFunction · 0.80
self_all_attentionFunction · 0.80
_create_taFunction · 0.80
raw_rnnFunction · 0.80

Calls

no outgoing calls

Tested by 15

fMethod · 0.64
testTensorArraysMethod · 0.64
zero_stateMethod · 0.64
_testRawRNNMethod · 0.64
testLoopStateMethod · 0.64
loop_fnMethod · 0.64
factoryMethod · 0.64