| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/mindrecord/include/ |
| D | shard_task_list.h | 55 TaskType task_type; member
203 inline void ShardTaskList::InsertTask(TaskType task_type, int shard_id, int group_id, in InsertTask()
213 inline void ShardTaskList::InsertTask(const int64_t &i, TaskType task_type, int shard_id, int group… in InsertTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pipeline/task/ |
| D | task.h | 43 explicit AsyncTask(TaskType task_type) : task_type_(task_type) {} in AsyncTask()
49 TaskType task_type() const { return task_type_; } in task_type() function
|
| D | device_task.h | 59 DeviceOpTask(std::shared_ptr<OpTaskContext> context, TaskType task_type) in DeviceOpTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/auto_grad/ |
| D | async_task_manager.h | 36 …explicit AsyncTaskMultiWorker(runtime::TaskType task_type) : comp_count_(0), task_type_(task_type)… in AsyncTaskMultiWorker()
39 runtime::TaskType task_type() const { return task_type_; } in task_type() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/ |
| D | mindrecord_op.cc | 211 auto task_type = task_content_ptr->first; in GetRowFromReader() local
237 … const mindrecord::json &columns_json, const mindrecord::TaskType task_type) { in LoadTensorRow()
|
| D | coco_op.h | 83 Builder &SetTask(const std::string &task_type) { in SetTask()
|
| D | voc_op.cc | 47 VOCOp::VOCOp(const TaskType &task_type, const std::string &task_mode, const std::string &folder_pat… in VOCOp()
62 VOCOp::VOCOp(const TaskType &task_type, const std::string &task_mode, const std::string &folder_pat… in VOCOp()
|
| D | coco_op.cc | 50 CocoOp::CocoOp(const TaskType &task_type, const std::string &image_folder_path, const std::string &… in CocoOp()
62 CocoOp::CocoOp(const TaskType &task_type, const std::string &image_folder_path, const std::string &… in CocoOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/forward/ |
| D | forward_task.h | 106 …:function<void(const FrontendOpRunInfoPtr &op_run_info, const runtime::KernelTaskType &task_type)>… in ViewKernelBackendTask()
|
| D | forward.cc | 349 … const runtime::KernelTaskType &task_type, bool enable_async) { in ForwardRunViewKernelTask()
364 runtime::KernelTaskType task_type, in CreateViewOpOutputs()
433 runtime::KernelTaskType task_type = GetViewOpTaskType(op_run_info->base_op_run_info.op_name); in ProcessViewOp() local
1002 runtime::KernelTaskType task_type) { in CreateViewOutputTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ir/datasetops/source/ |
| D | coco_node.cc | 89 CocoOp::TaskType task_type = CocoOp::TaskType::Detection; in Build() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ |
| D | cpu_device_context.cc | 84 runtime::KernelTaskPtr GetTaskByTaskType(const runtime::KernelTaskType &task_type, in GetTaskByTaskType()
531 bool CPUKernelExecutor::ExecuteKernelTask(const runtime::KernelTaskType &task_type, in ExecuteKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/mindrecord/meta/ |
| D | shard_task_list.cc | 186 TaskType task_type = TaskType::kCommonTask; in GetTaskByID() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/ |
| D | executor.cc | 137 enum TaskType task_type = task->type_; in WorkerLoop() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/mock/ |
| D | executor.cc | 195 enum TaskType task_type = task->type_; in WorkerLoop() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/ |
| D | executor.cc | 195 enum TaskType task_type = task->type_; in WorkerLoop() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/hardware/ |
| D | gpu_device_context.cc | 101 runtime::KernelTaskPtr GetTaskByTaskType(const runtime::KernelTaskType &task_type, in GetTaskByTaskType()
1078 bool GPUKernelExecutor::ExecuteKernelTask(const runtime::KernelTaskType &task_type, in ExecuteKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/dump/ |
| D | ascend_dump.cc | 146 uint64_t task_type = UnpackUint64Value(data_ptr + index); in ParseOverflowInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/hardware/ |
| D | device_context.h | 404 virtual bool ExecuteKernelTask(const runtime::KernelTaskType &task_type, in ExecuteKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/ |
| D | backend.cc | 1129 void MindRTBackend::RunViewKernelTaskAsyncImpl(const runtime::KernelTaskType &task_type, DeviceCont… in RunViewKernelTaskAsyncImpl()
1142 const runtime::KernelTaskType &task_type, bool enable_async) { in RunViewKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 807 void OpRunner::LaunchKernelTask(const runtime::KernelTaskType &task_type, DeviceContext *device_con… in LaunchKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | ge_kernel_executor.cc | 1202 bool GeKernelExecutor::ExecuteKernelTask(const runtime::KernelTaskType &task_type, in ExecuteKernelTask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/mindrecord/io/ |
| D | shard_reader.cc | 1523 auto task_type = std::get<0>(task); in ConsumerOneTask() local
|