Home
last modified time | relevance | path

Searched refs:kCpu (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dexecute.h39 …t Execute(std::shared_ptr<TensorOperation> op, MapTargetDevice device_type = MapTargetDevice::kCpu,
46 …t Execute(std::shared_ptr<TensorTransform> op, MapTargetDevice device_type = MapTargetDevice::kCpu,
53 …te(std::reference_wrapper<TensorTransform> op, MapTargetDevice device_type = MapTargetDevice::kCpu,
60 …explicit Execute(TensorTransform *op, MapTargetDevice device_type = MapTargetDevice::kCpu, uint32_…
68 MapTargetDevice device_type = MapTargetDevice::kCpu, uint32_t device_id = 0);
76 MapTargetDevice device_type = MapTargetDevice::kCpu, uint32_t device_id = 0);
84 MapTargetDevice device_type = MapTargetDevice::kCpu, uint32_t device_id = 0);
91 …ute(const std::vector<TensorTransform *> &ops, MapTargetDevice device_type = MapTargetDevice::kCpu,
Dconstants.h55 kCpu, ///< CPU Device. enumerator
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dexecute.cc129 if (device_type == MapTargetDevice::kCpu) { in Execute()
191 …{MapTargetDevice::kCpu, "kCpu"}, {MapTargetDevice::kGpu, "kGpu"}, {MapTargetDevice::kAscend310, "k… in operator ()()
204 if (device_type_ == MapTargetDevice::kCpu) { in operator ()()
287 …{MapTargetDevice::kCpu, "kCpu"}, {MapTargetDevice::kGpu, "kGpu"}, {MapTargetDevice::kAscend310, "k… in operator ()()
301 if (device_type_ == MapTargetDevice::kCpu) { // Case CPU in operator ()()
607 if (device_type_ == MapTargetDevice::kCpu) { in ParseTransforms()
626 if (device_type_ != MapTargetDevice::kCpu && device_type_ != MapTargetDevice::kAscend310 && in ValidateDevice()
Dvision.cc197 } else if (env == MapTargetDevice::kCpu) { in Parse()
279 } else if (env == MapTargetDevice::kCpu) { in Parse()
410 } else if (env == MapTargetDevice::kCpu) { in Parse()
852 } else if (env == MapTargetDevice::kCpu) { in Parse()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/
Dmap_op.cc90 MapTargetDevice prev_target = MapTargetDevice::kCpu; in GenerateWorkerJob()
94 MapTargetDevice target_device = MapTargetDevice::kCpu; in GenerateWorkerJob()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/oplib/
Doplib.cc55 constexpr auto kCpu = "CPU"; variable
96 } else if (imply_type_string == kCpu) { in RegOp()
/third_party/mindspore/tests/ut/cpp/dataset/
Dexecute_test.cc417 mindspore::dataset::Execute Transform(op_list, MapTargetDevice::kCpu); in TEST_F()