Searched refs:ascend310_info (Results 1 – 9 of 9) sorted by relevance
36 auto ascend310_info = device_infos[0]->Cast<Ascend310DeviceInfo>(); in AclModelOptions() local37 if (ascend310_info == nullptr) { in AclModelOptions()41 insert_op_cfg_path_ = ascend310_info->GetInsertOpConfigPath(); in AclModelOptions()42 input_format_ = ascend310_info->GetInputFormat(); in AclModelOptions()43 input_shape_map_ = ascend310_info->GetInputShapeMap(); in AclModelOptions()44 auto out_type = ascend310_info->GetOutputType(); in AclModelOptions()53 dynamic_batch_size_ = ascend310_info->GetDynamicBatchSize(); in AclModelOptions()54 precision_mode_ = ascend310_info->GetPrecisionMode(); in AclModelOptions()55 op_select_impl_mode_ = ascend310_info->GetOpSelectImplMode(); in AclModelOptions()56 fusion_switch_cfg_path_ = ascend310_info->GetFusionSwitchConfigPath(); in AclModelOptions()[all …]
142 auto ascend310_info = device_infos[0]->Cast<Ascend310DeviceInfo>(); in Resize() local143 MS_EXCEPTION_IF_NULL(ascend310_info); in Resize()144 ascend310_info->SetInputShape(input_shape_option); in Resize()
74 auto ascend310_info = std::make_shared<mindspore::Ascend310DeviceInfo>(); in ContextAutoSet() local75 ascend310_info->SetDeviceID(device_id); in ContextAutoSet()76 context->MutableDeviceInfo().emplace_back(ascend310_info); in ContextAutoSet()78 auto ascend310_info = std::make_shared<mindspore::Ascend310DeviceInfo>(); in ContextAutoSet() local79 ascend310_info->SetDeviceID(device_id); in ContextAutoSet()80 context->MutableDeviceInfo().emplace_back(ascend310_info); in ContextAutoSet()
104 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local105 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()106 auto device_id = ascend310_info->GetDeviceID(); in TEST_F()157 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local158 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()159 auto device_id = ascend310_info->GetDeviceID(); in TEST_F()205 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local206 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()207 auto device_id = ascend310_info->GetDeviceID(); in TEST_F()
62 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local63 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()64 ascend310_info->SetInsertOpConfigPath(aipp_path); in TEST_F()65 auto device_id = ascend310_info->GetDeviceID(); in TEST_F()110 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local111 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()112 ascend310_info->SetInsertOpConfigPath(aipp_path); in TEST_F()113 auto device_id = ascend310_info->GetDeviceID(); in TEST_F()
41 auto ascend310_info = context->MutableDeviceInfo()[0]->Cast<Ascend310DeviceInfo>(); in TEST_F() local42 ASSERT_TRUE(ascend310_info != nullptr); in TEST_F()48 ascend310_info->SetDynamicBatchSize(dynamic_batch_size); in TEST_F()49 ascend310_info->SetInputShapeMap(input_shape); in TEST_F()
79 auto ascend310_info = std::make_shared<mindspore::Ascend310DeviceInfo>(); in Ascend310DeviceInfoFromAscend310DeviceContext() local80 MS_CHECK_TRUE_RET(ascend310_info != nullptr, nullptr); in Ascend310DeviceInfoFromAscend310DeviceContext()81 ascend310_info->SetDeviceID(ascend310_context.device_info_.ascend310_device_info_.device_id_); in Ascend310DeviceInfoFromAscend310DeviceContext()82 return ascend310_info; in Ascend310DeviceInfoFromAscend310DeviceContext()
229 auto ascend310_info = std::make_shared<Ascend310DeviceInfo>(); in SetAclModelOptions() local230 ascend310_info->SetDeviceID(0); in SetAclModelOptions()231 model_context->MutableDeviceInfo().emplace_back(ascend310_info); in SetAclModelOptions()
981 auto ascend310_info = std::make_shared<Ascend310DeviceInfo>();982 model_context.MutableDeviceInfo().push_back(ascend310_info ); // set device target is ascend310983 ascend310_info->SetDeviceID(0); // set device id is 0984 ascend310_info->SetInsertOpConfigPath("./aipp.cfg"); // set aipp config file is ./aipp…