Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dtile_base.c32 …DoCopyData(input_data, output_data, src_dim_size, parameter->data_size_, parameter->multiples_[dim… in DoTileOneDimension()
39 …leOneDimension(input_data + in_pos * parameter->data_size_, output_data + out_pos * parameter->dat… in DoTileOneDimension()
53 …e_t dst_one_row_size = parameter->fast_stride_ * parameter->fast_multiple_ * parameter->data_size_; in TileSimple()
55 uint8_t *src = in_data + i * parameter->fast_stride_ * parameter->data_size_; in TileSimple()
56 … *dst = out_data + i * parameter->fast_stride_ * parameter->fast_multiple_ * parameter->data_size_; in TileSimple()
57 size_t offset = parameter->fast_stride_ * parameter->data_size_; in TileSimple()
Dtile_base.h38 size_t data_size_; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dreverse_fp32.cc42 data_size_ = in_tensors_.at(0)->ElementsNum(); in ReSize()
43 thread_sz_count_ = MSMIN(op_parameter_->thread_num_, data_size_); in ReSize()
48 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
66 tmp_ = reinterpret_cast<int *>(malloc(data_size_ * static_cast<int>(sizeof(int)))); in ReSize()
71 (void)memset(tmp_, 0, data_size_ * static_cast<int>(sizeof(int))); in ReSize()
85 for (int i = 0; i < data_size_; ++i) { in ReSize()
121 int count = MSMIN(thread_sz_stride_, data_size_ - task_id * thread_sz_stride_); in DoReverse()
Dfill_fp32.cc43 data_size_ = output->ElementsNum(); in ReSize()
44 thread_sz_count_ = MSMIN(thread_count_, data_size_); in ReSize()
46 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
52 int size = MSMIN(thread_sz_stride_, data_size_ - task_id * thread_sz_stride_); in DoFill()
Dfill_fp32.h42 int data_size_ = 0; variable
Dreverse_fp32.h48 int data_size_ = 0; variable
/third_party/cef/libcef_dll/wrapper/
Dcef_zip_archive.cc30 CefZipFile() : data_size_(0) {} in CefZipFile()
38 data_size_ = data_size; in Initialize()
42 data_size_ = 0; in Initialize()
49 virtual size_t GetDataSize() const override { return data_size_; } in GetDataSize()
53 data_.get(), data_size_, const_cast<CefZipFile*>(this))); in GetStreamReader()
60 size_t data_size_; member in __anonc5239a550111::CefZipFile
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/
Dstart_fl_job_kernel.h83 data_size_ = LongToInt(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "data_size")); in Init()
84 fl::worker::FLWorker::GetInstance().set_data_size(data_size_); in Init()
86 … << ", data_size: " << data_size_ << ". Request will be sent to server " << target_server_rank_; in Init()
108 req_fl_job_builder.add_data_size(data_size_); in BuildStartFLJobReq()
119 int data_size_; variable
Dupdate_model_kernel.h149 data_size_ = fl::worker::FLWorker::GetInstance().data_size(); in WeightingData()
153 data[i] *= data_size_; in WeightingData()
165 int data_size_; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Dfill_fp16.cc42 data_size_ = out_tensor->ElementsNum(); in ReSize()
43 thread_sz_count_ = MSMIN(thread_count_, data_size_); in ReSize()
48 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
53 int size = MSMIN(thread_sz_stride_, data_size_ - task_id * thread_sz_stride_); in DoFill()
Dfill_fp16.h42 int data_size_ = 0; variable
/third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ps_data/
Dps_data_channel.h34 data_size_(0) {} in PsDataChannel()
38 size_t data_size() const { return data_size_; } in data_size()
54 size_t data_size_; variable
Dps_data_channel.cc49 data_size_ = data_size; in set_data()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Darithmetic_self_fp32_coder.cc27 data_size_ = input_tensor_->ElementsNum(); in ReSize()
28 thread_sz_count_ = MSMIN(thread_num_, static_cast<int>(data_size_)); in ReSize()
30 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
65 …int size = MSMIN(thread_sz_stride_, static_cast<int>(data_size_ - kDefaultTaskId * thread_sz_strid… in DoCode()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dslice_cpu_kernel.cc62 data_size_ = size_pair->second; in InitKernel()
117 …static_cast<int8_t *>(input_addr) + data_size_ * slice_param_.shape_[1] * (start + slice_param_.be… in Launch()
118 auto dst = static_cast<int8_t *>(output_addr) + data_size_ * slice_param_.size_[1] * start; in Launch()
119 SliceSimpleDim2(src, dst, &slice_param_, data_size_, end - start); in Launch()
124 DoSliceNoParallel(input_addr, output_addr, &slice_param_, data_size_); in Launch()
Dstridedslice_cpu_kernel.cc125 data_size_ = type_pair->second.second; in InitSliceParam()
167 int inner_size = inner_ * data_size_; in RunTaskOnOuter()
182 int inner_size = inner_ * data_size_; in RunTaskOnSplitAxis()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dunsqueeze_int8.cc51 data_size_ = in_tensors_.at(0)->ElementsNum(); in ReSize()
52 thread_sz_count_ = MSMIN(thread_count_, data_size_); in ReSize()
54 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
59 size_t size = MSMIN(thread_sz_stride_, data_size_ - task_id * thread_sz_stride_); in DoUnsqueeze()
Darithmetic_self_int8.cc66 data_size_ = in_tensors_[0]->ElementsNum(); in ReSize()
67 thread_sz_count_ = MSMIN(thread_count_, static_cast<int>(data_size_)); in ReSize()
72 thread_sz_stride_ = UP_DIV(data_size_, thread_sz_count_); in ReSize()
87 int size = MSMIN(thread_sz_stride_, static_cast<int>(data_size_ - task_id * thread_sz_stride_)); in DoArithmeticSelf()
Dunsqueeze_int8.h47 int data_size_{0};
/third_party/mindspore/mindspore/ccsrc/debug/debugger/offline_debug/
Ddbg_services.h124 : data_size_(data_size), dtype_(dtype), shape_(shape) {} in TensorBaseData()
126 const uint64_t data_size() const { return data_size_; } in data_size()
129 uint64_t data_size_; member
138 : data_size_(data_size), in TensorStatData()
153 const uint64_t data_size() const { return data_size_; } in data_size()
168 uint64_t data_size_; member
/third_party/mindspore/mindspore/core/ir/
Dtensor.cc185 …explicit TensorDataImpl(const ShapeVector &shape) : ndim_(shape.size()), data_size_(SizeOf(shape))… in TensorDataImpl()
189 : ndim_(shape.size()), data_size_(SizeOf(shape)), data_(CopyData<T>(shape, data, data_len)) {} in TensorDataImpl()
192 … : ndim_(shape.size()), data_size_(SizeOf(shape)), data_(CopyData<T>(shape, data, data_type)) {} in TensorDataImpl()
196 : ndim_(shape.size()), data_size_(SizeOf(shape)), data_(NewData<T>(input, size)) {} in TensorDataImpl()
200 : ndim_(shape.size()), data_size_(SizeOf(shape)), data_(NewData<T>(scalar)) {} in TensorDataImpl()
202 ssize_t size() const override { return static_cast<ssize_t>(data_size_); } in size()
213 data_ = std::make_unique<T[]>(data_size_); in data()
235 return (ndim_ == ptr->ndim_) && (data_size_ == ptr->data_size_) && in equals()
236 std::equal(data_.get(), data_.get() + data_size_, ptr->data_.get()); in equals()
247 if (data_size_ == 0) { in ToString()
[all …]
/third_party/mindspore/mindspore/lite/examples/transfer_learning/src/
Dnet_runner.cc89 data_size_ = inputs[data_index_]->Size() / batch_size_; // in bytes in InitAndFigureInputs()
91 …std::cout << "data size: " << data_size_ << std::endl << "batch size: " << batch_size_ << std::end… in InitAndFigureInputs()
126 std::copy(data, data + data_size_, input_data + i * data_size_); in FillInputData()
162 if (data_size_ != 0) ds_.set_expected_data_size(data_size_); in InitDB()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dtile_base.cc71 tile_parameter_->data_size_ = sizeof(float); in ReSize()
73 tile_parameter_->data_size_ = sizeof(float) / 2; in ReSize()
144 tile_parameter_->data_size_ = lite::DataTypeSize(data_type); in Run()
/third_party/mindspore/mindspore/ccsrc/fl/worker/
Dfl_worker.h104 data_size_(0), in FLWorker()
140 int data_size_; variable
/third_party/mindspore/mindspore/ccsrc/cxx_api/
Dtypes.cc109 …: data_(nullptr), data_size_(0), name_(), type_(DataType::kTypeUnknown), shape_(), is_device_(fals… in TensorReferenceImpl()
113 …: data_(data), data_size_(data_len), name_(name), type_(type), shape_(shape), is_device_(is_device… in TensorReferenceImpl()
124 size_t DataSize() const override { return data_size_; } in DataSize()
129 … return std::make_shared<TensorReferenceImpl>(name_, type_, shape_, data_, data_size_, is_device_); in Clone()
134 size_t data_size_; member in mindspore::TensorReferenceImpl

12