Home
last modified time | relevance | path

Searched refs:NumOfElements (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Dconcatenate_op.cc42 output_shape = output_shape + inputs.at(0).NumOfElements(); in OutputShape()
47 (std::numeric_limits<uint64_t>::max() - output_shape) > prepend_->shape().NumOfElements(), in OutputShape()
49 output_shape = output_shape + prepend_->shape().NumOfElements(); in OutputShape()
55 (std::numeric_limits<uint64_t>::max() - output_shape) > append_->shape().NumOfElements(), in OutputShape()
56 … append parameter is too large to pend, got: " + std::to_string(append_->shape().NumOfElements())); in OutputShape()
57 output_shape = output_shape + append_->shape().NumOfElements(); in OutputShape()
Ddata_utils.cc205 for (int i = 0; i < input_shape.NumOfElements(); i++) { in FillHelper()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dngram_op.cc38 res.reserve(input->shape().NumOfElements()); // this should be more than enough in Compute()
39 offsets.reserve(1 + l_len_ + r_len_ + input->shape().NumOfElements()); in Compute()
87 …int32_t len_with_padding = inputs[0].NumOfElements() + std::min(n - 1, l_len_) + std::min(n - 1, r… in OutputShape()
/third_party/mindspore/tests/ut/cpp/dataset/
Dtensorshape_test.cc45 ASSERT_EQ(t.NumOfElements(), 120); in TEST_F()
62 ASSERT_EQ(t3.NumOfElements(), 0); in TEST_F()
65 ASSERT_EQ(t3.NumOfElements(), 0); in TEST_F()
75 ASSERT_EQ(t.NumOfElements(), 1); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/
Dtensor.h162 static_cast<dsize_t>(items.size()) == shape.NumOfElements(), in CreateFromVector()
329 dsize_t Size() const { return shape().NumOfElements(); } in Size()
333 if (data_end_ == nullptr) return type_.SizeInBytes() * shape_.NumOfElements(); in SizeInBytes()
716 …uchar *GetStringsBuffer() const { return data_ + kOffsetSize * shape_.NumOfElements() + kOffsetSiz… in GetStringsBuffer()
764 return TensorIterator<std::string_view>(data_, shape_.NumOfElements());
789 static_cast<dsize_t>(items.size()) == shape.NumOfElements(),
806 …dsize_t num_bytes = (kOffsetSize + 1) * (*out)->shape_.NumOfElements() + kOffsetSize + total_lengt…
Ddevice_tensor.cc44 if (!type.IsNumeric() && shape.NumOfElements() == 0) { in CreateEmpty()
73 if (!type.IsNumeric() && shape.NumOfElements() == 0) { in CreateFromDeviceMemory()
Dtensor_shape.cc42 dsize_t TensorShape::NumOfElements() const { in NumOfElements() function in mindspore::dataset::TensorShape
241 CHECK_FAIL_RETURN_UNEXPECTED(*flat_index < NumOfElements(), "Not a valid index"); in ToFlatIndex()
Dtensor.cc99 if (!type.IsNumeric() && shape.NumOfElements() == 0) { in CreateEmpty()
145 dsize_t min_length = (shape.NumOfElements() + 1) * kOffsetSize + shape.NumOfElements(); in CreateFromMemory()
232 …dsize_t num_bytes = (kOffsetSize) * (*out)->shape_.NumOfElements() + kOffsetSize + bytes_list.Byte… in CreateFromByteList()
484 if (shape.NumOfElements() == shape_.NumOfElements()) { in Reshape()
573 …(ind.at(0) + tensor->shape().NumOfElements() > shape().NumOfElements()) ? "[Tensor] incorrect inde… in InsertTensor()
835 uint64_t total_size = shape_.NumOfElements() * max_value; in GetDataAsNumpyStrings()
Dtensor_shape.h125 dsize_t NumOfElements() const;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/
Dplugin_op.cc52 CHECK_FAIL_RETURN_UNEXPECTED(in_row[ind]->shape().NumOfElements() == 1, in TensorRowToPlugin()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/
Drandom_data_op.cc291 int64_t size_in_bytes = new_shape->NumOfElements() * current_col.Type().SizeInBytes(); in CreateRandomRow()
Dtf_reader_op.cc455 if (cur_shape.known() && cur_shape.NumOfElements() != max_size) { in LoadBytesList()
458 … "\ntotal elements in shape received: " + std::to_string(cur_shape.NumOfElements()) + in LoadBytesList()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/
Dbatch_op.cc200 if (new_shape.NumOfElements() != 0) { in BatchRows()