Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dassert_cardinality_dataset_op.cc97 : DatasetIterator<Dataset>(params), num_elements_(0) {} in Iterator()
109 num_elements_++; in GetNextInternal()
111 if (*end_of_sequence && num_elements_ != dataset()->cardinality_) { in GetNextInternal()
115 ElementString(num_elements_), "."); in GetNextInternal()
118 num_elements_ > dataset()->cardinality_) { in GetNextInternal()
122 ElementString(num_elements_), "."); in GetNextInternal()
137 writer->WriteScalar(full_name("num_elements"), num_elements_)); in SaveInternal()
145 reader->ReadScalar(full_name("num_elements"), &num_elements_)); in RestoreInternal()
159 int64_t num_elements_; member in tensorflow::data::experimental::AssertCardinalityDatasetOp::Dataset::Iterator
Dlist_dataset_op.cc51 num_elements_(tensors_.size() / num_components), in Dataset()
66 std::make_unique<IndexSplitProvider>(num_elements_)); in MakeSplitProviders()
80 int64_t CardinalityInternal() const override { return num_elements_; } in CardinalityInternal()
83 return num_elements_; in CardinalityInternal()
136 std::make_shared<IndexSplitProvider>(dataset()->num_elements_); in Initialize()
185 int64 num_elements_; member in tensorflow::data::ListDatasetOp::Dataset
Dio_ops.cc378 (num_elements_)++; in GetNextInternal()
501 int64_t num_elements_; member in tensorflow::data::experimental::SaveDatasetV2Op::Dataset::Iterator
/external/webrtc/rtc_base/
Dswap_queue.h116 &num_elements_, size_t{0}, std::memory_order_relaxed); in Clear()
139 if (std::atomic_load_explicit(&num_elements_, std::memory_order_acquire) == in Insert()
152 &num_elements_, size_t{1}, std::memory_order_release); in Insert()
180 if (std::atomic_load_explicit(&num_elements_, std::memory_order_acquire) == in Remove()
192 std::atomic_fetch_sub_explicit(&num_elements_, size_t{1}, in Remove()
212 return std::atomic_load_explicit(&num_elements_, std::memory_order_acquire); in SizeAtLeast()
237 std::atomic<size_t> num_elements_{0};
/external/tensorflow/tensorflow/core/framework/
Dtensor_shape_test.cc305 int64_t num_elements() const { return num_elements_; } in num_elements()
344 int64_t num_elements_; member in tensorflow::__anon3caac34d0111::TensorShapeOld
407 num_elements_ = 1; in TensorShapeOld()
415 num_elements_ = 1; in TensorShapeOld()
421 TensorShapeOld::TensorShapeOld() : num_elements_(1) {} in TensorShapeOld()
425 num_elements_ = 1; in Clear()
431 num_elements_ *= size; in AddDim()
432 CHECK_LE(0, num_elements_); in AddDim()
433 CHECK_LE(num_elements_, kMaxElements); in AddDim()
445 num_elements_ *= size; in InsertDim()
[all …]
Dtensor_shape.h69 int64_t num_elements() const { return num_elements_; } in num_elements()
142 void set_num_elements(int64_t n) { num_elements_ = n; } in set_num_elements()
156 int64_t num_elements_; variable
692 num_elements_ = b.num_elements_; in TensorShapeRep()
705 num_elements_ = b.num_elements_; in TensorShapeRep()
720 num_elements_ = b.num_elements_;
735 num_elements_ = b.num_elements_;
Dmodel.cc553 if (num_elements_ == 0) { in ComputeSelfTime()
833 if (num_elements_ == 0) { in ComputeSelfTime()
1069 if (num_elements_ == 0 || inputs_.empty() || in RatioLocked()
1074 static_cast<double>(num_elements_); in RatioLocked()
1093 if (num_elements_ == 0 || inputs_.empty() || in InputTimeLocked()
1100 static_cast<double>(num_elements_); in InputTimeLocked()
1113 if (num_elements_ == 0 || inputs_.empty() || in OutputTimeLocked()
1126 static_cast<double>(num_elements_); in OutputTimeLocked()
1152 if (inputs_.empty() || num_elements_ == 0) { in TotalProcessingTimeLocked()
1160 static_cast<double>(num_elements_); in TotalProcessingTimeLocked()
[all …]
Dmodel.h180 num_elements_(0), in Node()
285 int64_t num_elements() const TF_LOCKS_EXCLUDED(mu_) { return num_elements_; } in num_elements()
329 num_elements_++; in record_element()
539 if (num_elements_ > 0) { in UpdateProcessingTimeEma()
541 static_cast<double>(num_elements_); in UpdateProcessingTimeEma()
688 std::atomic<int64_t> num_elements_; variable
/external/libvpx/test/
Dbuffer.h34 stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(nullptr) {} in Buffer()
41 padding_value_(0), stride_(0), raw_size_(0), num_elements_(0), in Buffer()
48 raw_size_(0), num_elements_(0), raw_buffer_(nullptr) {} in Buffer()
54 stride_(0), raw_size_(0), num_elements_(0), raw_buffer_(nullptr) {} in Buffer()
114 num_elements_ = stride_ * (top_padding_ + height_ + bottom_padding_); in Init()
115 raw_size_ = num_elements_ * sizeof(T); in Init()
127 raw_buffer_ = new (std::nothrow) T[num_elements_]; in Init()
147 int num_elements_; variable
283 for (int i = 0; i < num_elements_; ++i) { in SetPadding()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Drsqrt_builder.cc44 int num_elements_; member in tflite::delegates::hexagon::RsqrtOpBuilder
54 numerator.resize(num_elements_); in AddNumerator()
55 int flat_shape[] = {1, 1, 1, num_elements_}; in AddNumerator()
90 num_elements_ = batch_size * height_size * width_size * depth_size; in PopulateSubGraph()
91 int flat_shape[] = {1, 1, 1, num_elements_}; in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dtflite_driver.cc74 num_elements_ = 0; in Expectation()
80 num_elements_ = values.size(); in SetData()
81 data_.raw = new char[num_elements_ * sizeof(T)]; in SetData()
96 if (tensor_size != num_elements_) { in TypedCheck()
97 std::cerr << "Expected a tensor with " << num_elements_ in TypedCheck()
128 size_t num_elements_; member in tflite::testing::TfLiteDriver::Expectation
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops.cc43 num_elements_ = num_bytes / DataTypeSize(dtype_); in ScopedAllocatorOp()
60 context->allocate_output(0, {num_elements_}, &backing_tensor, attr); in Compute()
63 << " num_elements_ " << num_elements_ << " buffer " in Compute()
78 int64_t num_elements_; member in tensorflow::ScopedAllocatorOp
Dlist_kernels.h170 OP_REQUIRES_OK(c, c->GetAttr("num_elements", &num_elements_)); in TensorListStack()
181 if (num_elements_ != -1) { in Compute()
182 OP_REQUIRES(c, tensor_list->tensors().size() == num_elements_, in Compute()
184 "Operation expected a list with ", num_elements_, in Compute()
265 int num_elements_;
/external/tensorflow/tensorflow/core/kernels/data/
Dshuffle_dataset_op.cc215 if (num_elements_ == 0) { in GetNextInternal()
234 num_elements_--; in GetNextInternal()
277 writer->WriteScalar(this->full_name(kNumElements), num_elements_)); in SaveInternal()
325 reader->ReadScalar(this->full_name(kNumElements), &num_elements_)); in RestoreInternal()
391 << num_elements_ << " of " << BufferSizeString(); in FillBuffer()
425 if (slices_.size() > kMaxEpochsInBuffer && num_elements_ > 0) { in ShouldFillBuffer()
433 return num_elements_ < buffer_->size(); in ShouldFillBuffer()
456 if (num_elements_ == 0) { in AddToShuffleBuffer()
463 num_elements_++; in AddToShuffleBuffer()
488 int64_t num_elements_ TF_GUARDED_BY(mu_) = 0;
Dsparse_tensor_slice_dataset_op.cc94 num_elements_(params.dataset->sparse_tensor_.shape()[0]), in Iterator()
108 if (i_ == num_elements_) { in GetNextInternal()
207 const int64_t num_elements_; member in tensorflow::data::__anon0985c5ac0111::Dataset::Iterator
/external/tensorflow/tensorflow/lite/testing/
Dtflite_driver.cc121 num_elements_(0), in DataExpectation()
129 num_elements_ = values.size(); in SetData()
130 data_ = make_type_erased_array<T>(num_elements_); in SetData()
193 if (tensor_size != num_elements_) { in TypedCheck()
194 std::cerr << "Expected a tensor with " << num_elements_ in TypedCheck()
219 size_t num_elements_; member in tflite::testing::TfLiteDriver::DataExpectation
/external/cronet/third_party/protobuf/src/google/protobuf/
Dmap.h462 num_elements_(0),
633 std::swap(num_elements_, other->num_elements_);
673 num_elements_ = 0;
682 size_type size() const { return num_elements_; }
704 if (ResizeIfLoadIsOutOfRange(num_elements_ + 1)) {
722 ++num_elements_;
755 --num_elements_;
766 num_elements_, sizeof(Node));
1115 size_type num_elements_;
/external/protobuf/src/google/protobuf/
Dmap.h474 num_elements_(0),
645 std::swap(num_elements_, other->num_elements_);
685 num_elements_ = 0;
694 size_type size() const { return num_elements_; }
757 --num_elements_;
768 num_elements_, sizeof(Node));
779 if (ResizeIfLoadIsOutOfRange(num_elements_ + 1)) {
800 ++num_elements_;
1191 size_type num_elements_;
/external/OpenCL-CTS/test_conformance/subgroups/
Dsubhelpers.h1728 num_elements_(num_elements), test_params_(test_params)
1739 test<T, U>::run(device_, context_, queue_, num_elements_,
1752 int num_elements_;
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc1237 num_elements_ = PyObject_GetAttrString(py_vspace_, "num_elements_fn"); in Initialize()
1238 if (num_elements_ == nullptr) { in Initialize()
1269 Py_XDECREF(num_elements_); in ~PyVSpace()
1286 PyObject* result = PyEval_CallObject(num_elements_, arglist); in NumElements()
1427 PyObject* num_elements_; member in PyVSpace
/external/libaom/third_party/googletest/src/googlemock/include/gmock/
Dgmock-matchers.h3498 : num_elements_(num_elements),
3500 matched_(num_elements_ * num_matchers_, 0) {}
3502 size_t LhsSize() const { return num_elements_; }
3525 size_t num_elements_;
/external/cronet/third_party/googletest/src/googlemock/include/gmock/
Dgmock-matchers.h3492 : num_elements_(num_elements),
3494 matched_(num_elements_ * num_matchers_, 0) {}
3496 size_t LhsSize() const { return num_elements_; }
3519 size_t num_elements_;
/external/googletest/googlemock/include/gmock/
Dgmock-matchers.h3327 : num_elements_(num_elements),
3329 matched_(num_elements_* num_matchers_, 0) {
3332 size_t LhsSize() const { return num_elements_; }
3355 size_t num_elements_;
/external/fmtlib/test/gmock/
Dgmock.h7676 : num_elements_(num_elements),
7678 matched_(num_elements_* num_matchers_, 0) {
7681 size_t LhsSize() const { return num_elements_; }
7704 size_t num_elements_;
/external/rust/crates/grpcio-sys/grpc/third_party/cares/cares/test/gmock-1.8.0/gmock/
Dgmock.h8079 : num_elements_(num_elements),
8081 matched_(num_elements_* num_matchers_, 0) {
8084 size_t LhsSize() const { return num_elements_; }
8107 size_t num_elements_;