Home
last modified time | relevance | path

Searched defs:device_type (Results 1 – 25 of 94) sorted by relevance

1234

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/
Dfactory.h47 …void RegDelegate(const mindspore::DeviceType &device_type, const std::string &provider, DelegateCr… in RegDelegate()
57 void UnRegDelegate(const mindspore::DeviceType &device_type, const std::string &provider) { in UnRegDelegate()
63 T GetDelegate(const mindspore::DeviceType &device_type, const std::string &provider, in GetDelegate()
84 DelegateRegistrar(const mindspore::DeviceType &device_type, const std::string &provider, in DelegateRegistrar()
91 #define REG_DELEGATE(device_type, provider, creator) … argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dgraph_kernel_info.h45 void Register(const std::string &device_type, GraphKernelInfoCreator &&creator) { in Register()
51 std::shared_ptr<GraphKernelInfo> GetGraphKernelInfo(const std::string &device_type) { in GetGraphKernelInfo()
67 GraphKernelInfoRegister(const std::string &device_type, GraphKernelInfoCreator &&creator) { in GraphKernelInfoRegister()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/
Dgraph_kernel_builder_manager.cc26 void GraphKernelBuildManager::Register(const std::string &device_type, bool is_dynamic, in Register()
34 …<GraphKernelBuilder> GraphKernelBuildManager::GetGraphKernelBuilder(const std::string &device_type, in GetGraphKernelBuilder()
Dgraph_kernel_builder_manager.h43 …GraphKernelBuildRegister(const std::string &device_type, bool is_dynamic, GraphKernelBuildCreator … in GraphKernelBuildRegister()
/third_party/FreeBSD/sys/dev/usb/implementation/
Dusb_init.h53 typedef enum device_type { enum
64 } device_type; typedef
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/
Dfactory.cc33 std::shared_ptr<GraphCell::GraphImpl> GraphImplFactory::Create(enum DeviceType device_type) { in Create()
60 std::shared_ptr<ModelImpl> ModelImplFactory::Create(enum DeviceType device_type) { in Create()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/distributed/collective/
Dcollective_manager.cc156 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in Initialize() local
219 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in InitializeDummyCommLib() local
233 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in FinalizeDummyCommLib() local
512 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in InitDeviceCommLib() local
600 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in CreateSimulationGroup() local
627 std::string device_type = MsContext::GetInstance()->get_param<std::string>(MS_CTX_DEVICE_TARGET); in GetCommunicatorInitTimeout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/
Dexecute.cc68 Status Execute::InitResource(MapTargetDevice device_type, uint32_t device_id) { in InitResource()
127 Execute::Execute(const std::shared_ptr<TensorOperation> &op, MapTargetDevice device_type, uint32_t … in Execute()
145 Execute::Execute(const std::shared_ptr<TensorTransform> &op, MapTargetDevice device_type, uint32_t … in Execute()
154 Execute::Execute(const std::reference_wrapper<TensorTransform> &op, MapTargetDevice device_type, ui… in Execute()
166 Execute::Execute(TensorTransform *op, MapTargetDevice device_type, uint32_t device_id) { in Execute()
176 …ute::Execute(const std::vector<std::shared_ptr<TensorOperation>> &ops, MapTargetDevice device_type, in Execute()
183 …ute::Execute(const std::vector<std::shared_ptr<TensorTransform>> &ops, MapTargetDevice device_type, in Execute()
193 …ecute(const std::vector<std::reference_wrapper<TensorTransform>> &ops, MapTargetDevice device_type, in Execute()
213 Execute::Execute(const std::vector<TensorTransform *> &ops, MapTargetDevice device_type, uint32_t d… in Execute()
/third_party/skia/third_party/externals/angle2/scripts/
Dupdate_extension_data.py98 def name_device(gpu, device_type): argument
114 def get_props_string(gpu, bot_os, device_os, device_type): argument
119 def collect_task_and_update_json(task_id, gpu, bot_os, device_os, device_type): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/somas/
Dsomas.h241 void Register(device::DeviceType device_type, SomasCreator &&creator) { in Register()
248 SomasPtr GetSomas(device::DeviceType device_type) { in GetSomas()
263 SomasRegister(device::DeviceType device_type, SomasCreator &&creator) { in SomasRegister()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dinsert_cast_cpu.cc56 … const AnfNodePtr &cast, const std::string &format, const TypeId &device_type, in SyncWeightNodeWithCast()
93 auto device_type = AnfAlgo::GetInputDeviceDataType(cnode, input_index); in InsertCast() local
153 auto device_type = AnfAlgo::GetPrevNodeOutputDeviceDataType(func_output, i); in InsertCastForGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dinsert_cast.cc61 const auto device_type = AnfAlgo::GetOutputDeviceDataType(cnode, output_idx); in InsertCastForMultipleOutput() local
104 const TypeId device_type = AnfAlgo::GetOutputDeviceDataType(cnode, 0); in InsertCastForOutput() local
165 …if (TypeId device_type = AnfAlgo::GetInputDeviceDataType(cnode, input_index); origin_type != devic… in InsertCastForInput() local
/third_party/mindspore/mindspore-src/source/tests/st/networks/
Dtest_cpu_lenet_enable_rdr.py18 def test_train(device_type): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/
Ddata_queue_node.cc27 …:DataQueueNode(std::shared_ptr<DatasetNode> child, std::string queue_name, std::string device_type, in DataQueueNode()
129 std::string device_type = json_obj["device_type"]; in from_json() local
/third_party/mindspore/mindspore-src/source/tests/st/data_transfer/
Dtest_tdt_data_transfer_enable_rdr.py32 def test_train(device_type): argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/python/src/
Dtensor_pybind.cc138 …reate_tensor(DataType data_type, const std::vector<int64_t> &shape, const std::string &device_type, in create_tensor()
150 MSTensorPtr create_tensor_by_tensor(const MSTensor &tensor, const std::string &device_type, int dev… in create_tensor_by_tensor()
160 MSTensorPtr create_tensor_by_numpy(const py::array &input, const std::string &device_type, int32_t … in create_tensor_by_numpy()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/c_api/
Dtype_c_private.h33 OH_AI_NNRTDeviceType device_type; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dkernel_builder.cc50 auto device_type = Callback::Instance()->GetTargetFromContext(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/
Dfactory.h56 std::shared_ptr<T> Create(enum DeviceType device_type) { in Create()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/
Dmulti_conv_split_pass.cc66 …auto device_type = (device_type_attr != nullptr) ? GetValue<int32_t>(device_type_attr) : kDeviceTy… in Process() local
/third_party/mesa3d/src/gallium/frontends/rusticl/mesa/pipe/
Ddevice.rs37 pub fn device_type(&self) -> pipe_loader_device_type { in device_type() method
/third_party/mesa3d/src/gallium/frontends/clover/api/
Ddevice.cpp48 clGetDeviceIDs(cl_platform_id d_platform, cl_device_type device_type, in clGetDeviceIDs()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/
Dspliter.cc76 auto device_type = in RecordGraphInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/acl/
Dacl_model.cc171 bool AclModel::CheckDeviceSupport(mindspore::DeviceType device_type) { in CheckDeviceSupport()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/graph/acl/
Dacl_graph_impl.cc246 bool AclGraphImpl::CheckDeviceSupport(mindspore::DeviceType device_type) { in CheckDeviceSupport()

1234