Searched refs:set_data_size (Results 1 – 5 of 5) sorted by relevance
88 void set_data_size(int data_size);
174 void FLWorker::set_data_size(int data_size) { data_size_ = data_size; } in set_data_size() function in mindspore::fl::worker::FLWorker
84 fl::worker::FLWorker::GetInstance().set_data_size(data_size_); in Init()
162 device_meta.set_data_size(IntToSize(data_size)); in CreateDeviceMetadata()
943 tensor_base->set_data_size((int64_t)tensor_stat.data_size); in AddTensorStatInfo()1050 tensor_base_item.set_data_size(0); in LoadTensorsBase()1058 tensor_base_item.set_data_size((int64_t)tensor->GetByteSize()); in LoadTensorsBase()