Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/kernels/
Dtensor_array.h156 tensors_(N) {} in TensorArray()
266 return strings::StrCat("TensorArray[", tensors_.size(), "]"); in DebugString()
278 *size = tensors_.size(); in Size()
304 *size = is_grad_ ? marked_size_ : tensors_.size(); in PackOrConcatSize()
339 tensors_.clear(); in ClearAndMarkClosed()
436 std::vector<TensorAndState> tensors_ GUARDED_BY(mu_);
445 if (index < 0 || (!dynamic_size_ && index_size >= tensors_.size())) { in LockedWriteOrAggregate()
448 index, " but array is not resizeable and size is: ", tensors_.size()); in LockedWriteOrAggregate()
452 if (index_size >= tensors_.capacity()) { in LockedWriteOrAggregate()
453 tensors_.reserve(2 * (index_size + 1)); in LockedWriteOrAggregate()
[all …]
Dtensor_array.cc89 if (tensors_.size() != rhs->tensors_.size()) { in CopyShapesFrom()
92 handle_.vec<string>()(1), " has size ", tensors_.size(), " but rhs ", in CopyShapesFrom()
93 rhs->handle_.vec<string>()(1), " has size ", rhs->tensors_.size()); in CopyShapesFrom()
95 for (std::size_t i = 0; i < tensors_.size(); ++i) { in CopyShapesFrom()
97 if (!rhs->tensors_[i].written) continue; in CopyShapesFrom()
100 tensors_[i].shape = rhs->tensors_[i].shape; in CopyShapesFrom()
105 tensors_[i].written = true; in CopyShapesFrom()
Dops_testutil.h81 gtl::STLDeleteElements(&tensors_); in ~OpsTestBase()
156 tensors_.push_back(input); in AddResourceInput()
214 tensors_.push_back(input); in AddInput()
240 std::vector<Tensor*> tensors_; variable
Dresize_area_op_test.cc45 tensors_.push_back(input); in SetRandomImageInput()
Dresize_bicubic_op_test.cc45 tensors_.push_back(input); in SetRandomImageInput()
/external/tensorflow/tensorflow/core/framework/
Dvariant_tensor_data.cc31 int VariantTensorData::tensors_size() const { return tensors_.size(); } in tensors_size()
34 return tensors_[index]; in tensors()
38 return tensors_; in tensors()
42 tensors_.emplace_back(); in add_tensors()
43 return &(tensors_[tensors_.size() - 1]); in add_tensors()
50 for (const auto& tensor : tensors_) { in ToProto()
61 tensors_.push_back(tmp); in FromProto()
87 for (const auto& t : tensors_) { in DebugString()
Dsession_state.h53 std::unordered_map<string, Tensor> tensors_; variable
82 std::unordered_map<string, TensorAndKey> tensors_ GUARDED_BY(lock_);
Dvariant_op_copy_test.cc62 void Encode(VariantTensorData* data) const { data->tensors_ = {stored}; } in Encode()
64 CHECK_EQ(1, data.tensors_.size()); in Decode()
65 stored = data.tensors_[0]; in Decode()
Dvariant_encode_decode.h60 data->tensors_.clear(); in EncodeVariantImpl()
61 data->tensors_.push_back(value); in EncodeVariantImpl()
Dvariant_test.cc145 void Encode(VariantTensorData* data) const { data->tensors_ = vec; } in Encode()
148 vec = data.tensors_; in Decode()
Dvariant_tensor_data.h83 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.cc60 : GraphDatasetBase(ctx), tensors_(std::move(tensors)) { in Dataset()
61 for (const Tensor& t : tensors_) { in Dataset()
90 components.reserve(tensors_.size()); in AsGraphDefInternal()
91 for (const Tensor& t : tensors_) { in AsGraphDefInternal()
109 n_(params.dataset->tensors_[0].dim_size(0)) {} in Iterator()
117 out_tensors->reserve(dataset()->tensors_.size()); in GetNextInternal()
118 for (int i = 0; i < dataset()->tensors_.size(); ++i) { in GetNextInternal()
119 const Tensor& t = dataset()->tensors_[i]; in GetNextInternal()
153 const std::vector<Tensor> tensors_; member in tensorflow::__anon51dd03970111::TensorSliceDatasetOp::Dataset
Dtensor_dataset_op.cc49 : GraphDatasetBase(ctx), tensors_(std::move(tensors)) { in Dataset()
50 for (const Tensor& t : tensors_) { in Dataset()
73 components.reserve(tensors_.size()); in AsGraphDefInternal()
74 for (const Tensor& t : tensors_) { in AsGraphDefInternal()
97 *out_tensors = dataset()->tensors_; in GetNextInternal()
127 const std::vector<Tensor> tensors_; member in tensorflow::__anon601612280111::TensorDatasetOp::Dataset
/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/contrib/lite/
Dinterpreter.cc87 tensors_.reserve(kSlotsToReserve); in Interpreter()
450 int base_index = tensors_.size(); in AddTensors()
452 tensors_.resize(tensors_.size() + tensors_to_add); in AddTensors()
453 for (int i = base_index; i < tensors_.size(); i++) { in AddTensors()
454 memset(&tensors_[i], 0, sizeof(tensors_[i])); in AddTensors()
456 context_.tensors = tensors_.data(); in AddTensors()
457 context_.tensors_size = tensors_.size(); in AddTensors()
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()
57 void AddTensors(int count) { tensors_.resize(count + tensors_.size()); } in AddTensors()
67 std::vector<TfLiteTensor> tensors_; member in tflite::__anonab6bad3f0111::SimpleTestGraph
Darena_planner_test.cc83 tensors_.push_back(TfLiteTensor()); in TestGraph()
86 tensors_.back().allocation_type = kTfLiteArenaRw; in TestGraph()
87 tensors_.back().bytes = (i + 1) * 3; in TestGraph()
100 std::vector<TfLiteTensor>* tensors() { return &tensors_; } in tensors()
106 std::vector<TfLiteTensor> tensors_; member in tflite::__anona09efff80111::TestGraph
Dinterpreter.h307 std::vector<TfLiteTensor> tensors_; variable
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_launch_op.cc79 std::unordered_map<void*, Tensor> tensors_; member in tensorflow::XlaAllocator
106 tensors_[data] = t; in Allocate()
113 tensors_[data] = *t; in RegisterArgument()
120 if (tensors_.erase(mem->opaque()) == 0) { in Deallocate()
132 auto it = tensors_.find(ptr); in MakeTensorFromBuffer()
133 if (it == tensors_.end()) { in MakeTensorFromBuffer()
/external/tensorflow/tensorflow/contrib/lite/kernels/
Dtest_util.cc53 int id = tensors_.size(); in AddTensor()
96 tensors_.push_back(CreateTensor(builder_, builder_.CreateVector<int>(t.shape), in AddTensor()
159 auto tensors = builder_.CreateVector(tensors_); in BuildInterpreter()
Dtest_util.h215 std::vector<flatbuffers::Offset<Tensor>> tensors_; variable
/external/tensorflow/tensorflow/contrib/lite/tools/
Dverifier_test.cc64 tensors_.push_back(CreateTensorDirect(builder_, &shape, type, buffer_index, in AddTensor()
83 builder_, builder_.CreateVector(tensors_), in FinishModel()
105 std::vector<Offset<Tensor>> tensors_; member in tflite::TfLiteFlatbufferModelBuilder
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers.cc737 size_t InMemoryRunStepResponse::num_tensors() const { return tensors_.size(); } in num_tensors()
740 return tensors_[i].first; in tensor_name()
745 *out_tensor = tensors_[i].second; in TensorValue()
757 tensors_.emplace_back(name, tensor); in AddTensorFromRunGraphResponse()

12