| /third_party/mindspore/mindspore/lite/src/ |
| D | inner_context.cc | 105 cpu_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = NO_BIND; in SetContextDevice()
160 … device_ctx.device_info_.npu_device_info_.frequency_ != hiai::AiModelDescription_Frequency_LOW && in Init()
161 …device_ctx.device_info_.npu_device_info_.frequency_ != hiai::AiModelDescription_Frequency_MEDIUM && in Init()
162 … device_ctx.device_info_.npu_device_info_.frequency_ != hiai::AiModelDescription_Frequency_HIGH && in Init()
163 …device_ctx.device_info_.npu_device_info_.frequency_ != hiai::AiModelDescription_Frequency_EXTREME)… in Init()
165 << device_ctx.device_info_.npu_device_info_.frequency_; in Init()
166 … device_ctx.device_info_.npu_device_info_.frequency_ = hiai::AiModelDescription_Frequency_HIGH; in Init()
270 device_info = &device.device_info_; in GetCpuDeviceInfo()
318 …return device.device_type_ == DT_CPU && (device.device_info_.cpu_device_info_.cpu_bind_mode_ < NO_… in IsCpuBindModeInvalid()
319 … device.device_info_.cpu_device_info_.cpu_bind_mode_ > MID_CPU); in IsCpuBindModeInvalid()
[all …]
|
| /third_party/mindspore/mindspore/lite/java/native/runtime/ |
| D | ms_config.cpp | 32 cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = mindspore::lite::NO_BIND; in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
35 cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = mindspore::lite::HIGHER_CPU; in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
38 cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = mindspore::lite::MID_CPU; in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
46 cpu_device_ctx.device_info_.cpu_device_info_.enable_float16_ = true; in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
56 gpu_device_ctx.device_info_.gpu_device_info_.enable_float16_ = true; in Java_com_mindspore_lite_config_MSConfig_createMSConfig()
|
| /third_party/mindspore/mindspore/lite/src/common/ |
| D | context_util.cc | 42 cpu_info->SetEnableFP16(cpu_context.device_info_.cpu_device_info_.enable_float16_); in CPUDeviceInfoFromCPUDeviceContext()
54 gpu_info->SetEnableFP16(gpu_context.device_info_.gpu_device_info_.enable_float16_); in GPUDeviceInfoFromGPUDeviceContext()
55 gpu_info->SetDeviceID(gpu_context.device_info_.gpu_device_info_.gpu_device_id_); in GPUDeviceInfoFromGPUDeviceContext()
68 npu_info->SetFrequency(npu_context.device_info_.npu_device_info_.frequency_); in NPUDeviceInfoFromNPUDeviceContext()
81 ascend310_info->SetDeviceID(ascend310_context.device_info_.ascend310_device_info_.device_id_); in Ascend310DeviceInfoFromAscend310DeviceContext()
130 ms_context->SetThreadAffinity(device_context.device_info_.cpu_device_info_.cpu_bind_mode_); in MSContextFromContext()
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | meta_tensor.h | 138 DeviceInfo device_info() const { return device_info_; } in device_info()
141 void set_device_info(const DeviceInfo &device_info) { device_info_ = device_info; } in set_device_info()
198 DeviceInfo device_info_; variable
|
| D | meta_tensor.cc | 51 device_info_ = meta_tensor.device_info(); in operator =()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/ |
| D | network_test.cc | 102 context->device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = lite::NO_BIND; in TEST_F()
127 context->device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = lite::NO_BIND; in TEST_F()
149 context.device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = lite::NO_BIND; in TEST_F()
168 context.device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = lite::NO_BIND; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_delegate.cc | 74 device_info_ = gpu_info; in Init()
107 lite::SetCudaDevice(device_info_); in Init()
119 lite::SetCudaDevice(device_info_); in Build()
185 …TensorRTSubGraph(ops, in_tensors, out_tensors, context_, device_info_, runtime_, support_hw_resize… in CreateTensorRTGraph()
|
| D | tensorrt_subgraph.h | 38 device_info_(device_info), in TensorRTSubGraph()
85 std::shared_ptr<GPUDeviceInfo> device_info_{nullptr};
|
| D | tensorrt_delegate.h | 56 std::shared_ptr<GPUDeviceInfo> device_info_{nullptr};
|
| D | tensorrt_subgraph.cc | 103 if (device_info_->GetEnableFP16() && runtime_->GetBuilder()->platformHasFastFp16()) { in SetDeviceConfig()
301 lite::SetCudaDevice(device_info_); in Prepare()
427 lite::SetCudaDevice(device_info_); in Execute()
|
| D | tensorrt_utils.h | 76 void SetCudaDevice(std::shared_ptr<GPUDeviceInfo> device_info_);
|
| D | tensorrt_utils.cc | 264 void SetCudaDevice(std::shared_ptr<GPUDeviceInfo> device_info_) { in SetCudaDevice() argument
270 int set_device_id = static_cast<int>(device_info_->GetDeviceID()); in SetCudaDevice()
|
| /third_party/mindspore/mindspore/lite/test/st/ |
| D | mindrt_parallel_runtime_test.cc | 69 gpu_device_ctx.device_info_.gpu_device_info_.enable_float16_ = false; in TEST_F()
120 cpu_device_ctx.device_info_.cpu_device_info_.enable_float16_ = true; in TEST_F()
|
| D | sub_graph_test.cc | 329 cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = lite::MID_CPU; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/include/ |
| D | context.h | 59 DeviceInfo device_info_; member
|
| /third_party/mindspore/mindspore/lite/include/ |
| D | context.h | 62 DeviceInfo device_info_; member
|
| /third_party/mindspore/mindspore/lite/tools/benchmark/ |
| D | benchmark.cc | 125 … cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = CpuBindMode(flags_->cpu_bind_mode_); in InitContext()
127 cpu_device_ctx.device_info_.cpu_device_info_.cpu_bind_mode_ = NO_BIND; in InitContext()
129 cpu_device_ctx.device_info_.cpu_device_info_.enable_float16_ = flags_->enable_fp16_; in InitContext()
133 gpu_device_ctx.device_info_.gpu_device_info_.enable_float16_ = flags_->enable_fp16_; in InitContext()
139 npu_device_ctx.device_info_.npu_device_info_.frequency_ = AiModelDescription_Frequency_HIGH; in InitContext()
|
| /third_party/mindspore/mindspore/lite/examples/transfer_learning/src/ |
| D | net_runner.cc | 77 context.device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = mindspore::lite::NO_BIND; in InitAndFigureInputs()
78 context.device_list_[0].device_info_.cpu_device_info_.enable_float16_ = enable_fp16_; in InitAndFigureInputs()
|
| /third_party/mindspore/mindspore/lite/examples/train_lenet/src/ |
| D | net_runner.cc | 148 context.device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = mindspore::lite::NO_BIND; in InitAndFigureInputs()
149 context.device_list_[0].device_info_.cpu_device_info_.enable_float16_ = enable_fp16_; in InitAndFigureInputs()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/ |
| D | benchmark.cc | 126 context->device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_); in main()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/benchmark/ |
| D | benchmark.cc | 126 context->device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_); in main()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/ |
| D | infer_test.cc | 246 context->device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = lite::NO_BIND; in TEST_F()
|
| /third_party/mindspore/mindspore/lite/tools/benchmark_train/ |
| D | net_train.cc | 407 …context.device_list_[0].device_info_.cpu_device_info_.cpu_bind_mode_ = FlagToBindMode(flags_->cpu_… in CreateAndRunNetwork()
408 context.device_list_[0].device_info_.cpu_device_info_.enable_float16_ = flags_->enable_fp16_; in CreateAndRunNetwork()
|