Home
last modified time | relevance | path

Searched refs:tensors_ (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dunbatch_dataset_op.cc109 out_tensors->reserve(tensors_.size()); in GetNextInternal()
110 for (int i = 0; i < tensors_.size(); ++i) { in GetNextInternal()
111 out_tensors->emplace_back(ctx->allocator({}), tensors_[i].dtype(), in GetNextInternal()
114 &tensors_[i], &out_tensors->back(), current_index_)); in GetNextInternal()
122 tensors_.clear(); in GetNextInternal()
124 input_impl_->GetNext(ctx, &tensors_, end_of_sequence)); in GetNextInternal()
126 for (size_t i = 0; i < tensors_.size(); ++i) { in GetNextInternal()
127 if (tensors_[i].dims() == 0) { in GetNextInternal()
132 if (tensors_[i].dim_size(0) != tensors_[0].dim_size(0)) { in GetNextInternal()
136 tensors_[0].dim_size(0), " but component ", i, in GetNextInternal()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.cc27 int VariantTensorData::tensors_size() const { return tensors_.size(); } in tensors_size()
30 return tensors_[index]; in tensors()
34 return tensors_; in tensors()
38 tensors_.emplace_back(); in add_tensors()
39 return &(tensors_[tensors_.size() - 1]); in add_tensors()
44 tensors_.emplace_back(std::forward<TensorConstructorArgs>(args)...); in add_tensor()
45 return &tensors_.back(); in add_tensor()
52 for (const auto& tensor : tensors_) { in ToProto()
64 tensors_.push_back(tmp); in FromProto()
75 tensors_.push_back(tmp); in FromConstProto()
[all …]
Dsession_state.h53 std::unordered_map<string, Tensor> tensors_; variable
80 return tensors_.empty(); in empty()
88 std::unordered_map<string, TensorAndKey> tensors_ GUARDED_BY(lock_);
Dvariant_op_copy_test.cc63 void Encode(VariantTensorData* data) const { data->tensors_ = {stored}; } in Encode()
65 CHECK_EQ(1, data.tensors_.size()); in Decode()
66 stored = data.tensors_[0]; in Decode()
Dvariant_encode_decode.h61 data->tensors_.clear(); in EncodeVariantImpl()
62 data->tensors_.push_back(value); in EncodeVariantImpl()
Dvariant_test.cc145 void Encode(VariantTensorData* data) const { data->tensors_ = vec; } in Encode()
148 vec = std::move(data.tensors_); in Decode()
Dvariant_tensor_data.h92 std::vector<Tensor> tensors_; variable
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array.h157 tensors_(N) {} in TensorArray()
267 return strings::StrCat("TensorArray[", tensors_.size(), "]"); in DebugString()
279 *size = tensors_.size(); in Size()
305 *size = is_grad_ ? marked_size_ : tensors_.size(); in PackOrConcatSize()
342 tensors_.clear(); in ClearAndMarkClosed()
439 std::vector<TensorAndState> tensors_ GUARDED_BY(mu_);
448 if (index < 0 || (!dynamic_size_ && index_size >= tensors_.size())) { in LockedWriteOrAggregate()
451 index, " but array is not resizeable and size is: ", tensors_.size()); in LockedWriteOrAggregate()
455 if (index_size >= tensors_.capacity()) { in LockedWriteOrAggregate()
456 tensors_.reserve(2 * (index_size + 1)); in LockedWriteOrAggregate()
[all …]
Dtensor_array.cc91 if (tensors_.size() != rhs->tensors_.size()) { in CopyShapesFrom()
94 handle_.vec<string>()(1), " has size ", tensors_.size(), " but rhs ", in CopyShapesFrom()
95 rhs->handle_.vec<string>()(1), " has size ", rhs->tensors_.size()); in CopyShapesFrom()
97 for (std::size_t i = 0; i < tensors_.size(); ++i) { in CopyShapesFrom()
99 if (!rhs->tensors_[i].written) continue; in CopyShapesFrom()
103 tensors_[i].shape = *shape_to_prepend; in CopyShapesFrom()
104 tensors_[i].shape.AppendShape(rhs->tensors_[i].shape); in CopyShapesFrom()
106 tensors_[i].shape = rhs->tensors_[i].shape; in CopyShapesFrom()
112 tensors_[i].written = true; in CopyShapesFrom()
Dops_testutil.h81 gtl::STLDeleteElements(&tensors_); in ~OpsTestBase()
155 tensors_.push_back(input); in AddResourceInput()
213 tensors_.push_back(input); in AddInput()
239 std::vector<Tensor*> tensors_; variable
/external/tensorflow/tensorflow/core/common_runtime/
Dsession_state.cc25 auto it = tensors_.find(handle); in GetTensor()
26 if (it == tensors_.end()) { in GetTensor()
36 if (!tensors_.insert({handle, tensor}).second) { in AddTensor()
45 if (tensors_.erase(handle) == 0) { in DeleteTensor()
59 if (!tensors_.insert({name, tk}).second) { in AddTensor()
69 if (!tensors_.empty()) { in SaveTensors()
74 auto it = tensors_.find(op_name); in SaveTensors()
75 if (it != tensors_.end()) { in SaveTensors()
/external/tensorflow/tensorflow/core/kernels/data/
Dtensor_slice_dataset_op.cc67 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
68 for (const Tensor& t : tensors_) { in Dataset()
95 int64 Cardinality() const override { return tensors_[0].dim_size(0); } in Cardinality()
102 components.reserve(tensors_.size()); in AsGraphDefInternal()
103 for (const Tensor& t : tensors_) { in AsGraphDefInternal()
127 n_(params.dataset->tensors_[0].dim_size(0)) {} in Iterator()
135 out_tensors->reserve(dataset()->tensors_.size()); in GetNextInternal()
136 for (int i = 0; i < dataset()->tensors_.size(); ++i) { in GetNextInternal()
137 const Tensor& t = dataset()->tensors_[i]; in GetNextInternal()
177 const std::vector<Tensor> tensors_; member in tensorflow::data::__anon55e027fa0111::TensorSliceDatasetOp::Dataset
Dtensor_dataset_op.cc50 : DatasetBase(DatasetContext(ctx)), tensors_(std::move(tensors)) { in Dataset()
51 for (const Tensor& t : tensors_) { in Dataset()
77 components.reserve(tensors_.size()); in AsGraphDefInternal()
78 for (const Tensor& t : tensors_) { in AsGraphDefInternal()
107 *out_tensors = dataset()->tensors_; in GetNextInternal()
142 const std::vector<Tensor> tensors_; member in tensorflow::data::__anon927a684b0111::TensorDatasetOp::Dataset
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.h115 for (const auto& entry : tensors_) { in ~ServerContext()
128 tensors_.emplace(RemoteTensorHandleInternal(operation_id, i), in AddOperationOutputs()
136 auto iter = tensors_.find(remote_handle); in GetTensorHandle()
137 if (iter == tensors_.end()) { in GetTensorHandle()
150 auto iter = tensors_.find(remote_handle); in DeleteTensorHandle()
151 if (iter == tensors_.end()) { in DeleteTensorHandle()
158 tensors_.erase(iter); in DeleteTensorHandle()
186 RemoteTensorHandleMap tensors_ GUARDED_BY(tensors_mu_);
/external/tensorflow/tensorflow/lite/experimental/micro/
Dmicro_interpreter.cc79 tensors_ = subgraph_->tensors(); in MicroInterpreter()
82 context_.tensors_size = tensors_->Length(); in MicroInterpreter()
88 const auto* tensor = tensors_->Get(tensor_index); in MicroInterpreter()
98 sizeof(int) * tensors_->Length(), sizeof(int))); in MicroInterpreter()
100 sizeof(int) * tensors_->Length(), sizeof(int))); in MicroInterpreter()
101 for (int i = 0; i < tensors_->Length(); ++i) { in MicroInterpreter()
121 const auto* tensor = tensors_->Get(tensor_index); in MicroInterpreter()
134 for (int i = 0; i < tensors_->Length(); ++i) { in MicroInterpreter()
135 const auto* tensor = tensors_->Get(i); in MicroInterpreter()
Dmicro_interpreter.h62 const flatbuffers::Vector<flatbuffers::Offset<Tensor>>* tensors_; variable
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h161 size_t tensors_size() const { return tensors_.size(); } in tensors_size()
174 std::vector<TfLiteTensor>& tensors() { return tensors_; } in tensors()
249 TfLiteTensor* t = &tensors_[tensor_index]; in EnsureTensorDataIsReadable()
347 std::vector<TfLiteTensor> tensors_; variable
441 const size_t required_capacity = tensors_.size() + kTensorsCapacityHeadroom; in EnsureTensorsVectorCapacity()
442 if (required_capacity > tensors_.capacity()) { in EnsureTensorsVectorCapacity()
443 tensors_.reserve(required_capacity); in EnsureTensorsVectorCapacity()
444 context_->tensors = tensors_.data(); in EnsureTensorsVectorCapacity()
Dsubgraph.cc160 tensors_.reserve(kTensorsReservedCapacity); in Subgraph()
307 TfLiteTensor* tensor = &tensors_[tensor_index]; in ReplaceNodeSubsetsWithDelegateKernels()
502 for (auto& tensor : tensors_) { in ResetVariableTensors()
709 TfLiteTensor* tensor = &tensors_[tensor_index]; in Invoke()
779 const size_t base_index = tensors_.size(); in AddTensors()
781 tensors_.resize(tensors_.size() + tensors_to_add); in AddTensors()
782 for (size_t i = base_index; i < tensors_.size(); i++) { in AddTensors()
783 memset(&tensors_[i], 0, sizeof(tensors_[i])); in AddTensors()
784 tensors_[i].buffer_handle = kTfLiteNullBufferHandle; in AddTensors()
786 context_->tensors = tensors_.data(); in AddTensors()
[all …]
/external/tensorflow/tensorflow/core/util/
Dtensor_slice_reader.cc175 ss_slice, &tensors_); in LoadShard()
192 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in FindTensorSlice()
199 TensorSliceReader::~TensorSliceReader() { gtl::STLDeleteValues(&tensors_); } in ~TensorSliceReader()
204 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in HasTensor()
209 tss = gtl::FindPtrOrNull(tensors_, name); in HasTensor()
231 const TensorSliceSet* tss = gtl::FindPtrOrNull(tensors_, name); in GetTensor()
Dtensor_slice_reader.h96 return tensors_; in Tensors()
135 mutable std::unordered_map<string, TensorSliceSet*> tensors_; variable
/external/tensorflow/tensorflow/lite/
Dgraph_info_test.cc42 size_t num_tensors() const override { return tensors_.size(); } in num_tensors()
45 TfLiteTensor* tensor(size_t index) override { return &tensors_[index]; } in tensor()
58 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors()
68 std::vector<TfLiteTensor> tensors_; member in tflite::__anon6e38e6c20111::SimpleTestGraph
Darena_planner_test.cc85 tensors_.push_back(TfLiteTensor()); in TestGraph()
88 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
89 tensors_.back().bytes = (i + 1) * 3; in TestGraph()
102 std::vector<TfLiteTensor>* tensors() { return &tensors_; } in tensors()
113 std::swap(tensors_, other->tensors_); in Swap()
121 std::vector<TfLiteTensor> tensors_; member in tflite::__anonabd56e9b0111::TestGraph
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h363 const int id = tensors_.size(); in AddTensorPerChannelQuant()
372 tensors_.push_back( in AddTensorPerChannelQuant()
383 int id = tensors_.size();
434 tensors_.push_back(CreateTensor(builder_,
472 std::vector<flatbuffers::Offset<Tensor>> tensors_; variable
/external/tensorflow/tensorflow/lite/tools/
Dverifier_test.cc83 tensors_.push_back(CreateTensorDirect(builder_, /*shape=*/nullptr, type, in AddTensor()
88 tensors_.push_back(CreateTensorDirect(builder_, &shape, type, buffer_index, in AddTensor()
108 builder_, builder_.CreateVector(tensors_), in FinishModel()
133 std::vector<Offset<Tensor>> tensors_; member in tflite::TfLiteFlatbufferModelBuilder
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc796 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors()
799 return tensors_[i].first; in tensor_name()
804 *out_tensor = tensors_[i].second; in TensorValue()
816 tensors_.emplace_back(name, tensor); in AddTensorFromRunGraphResponse()

12