Home
last modified time | relevance | path

Searched defs:output_kernel_tensors (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/device/
Dhccl_adapter_test.cc108 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
133 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
159 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
185 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
210 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
235 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
261 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
287 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
313 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
335 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(all… in TEST_F() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Didentity.cc66 auto output_kernel_tensors = {launch_device_address->kernel_tensor().get()}; in IdentityCustomizeCallWithoutContigous() local
114 auto output_kernel_tensors = {output_address->kernel_tensor().get()}; in IdentityCustomizeCall() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/host/
Dhost_kernel_build.cc39 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in HostOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/
Dhccl_kernel_build.cc46 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in HcclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/rts/
Dcondition_gather.cc28 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in Init() local
Dcondition_switch.cc28 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in Init() local
Dreshape_ext.cc29 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in Init() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc43 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/
Daclnn_kernel_build.cc45 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in AclnnOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/
Ddynamic_akg_gpu_kernel_build.cc40 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/
Ddynamic_akg_cpu_kernel_build.cc34 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in SetKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dcpu_pass_utils.cc83 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(cas… in AddCastOpNodeToGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/bisheng/
Dbisheng_kernel_build.cc74 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(cno… in BiShengOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_kernel_build.cc121 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(ker… in CreateGPUKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/
Dinternal_kernel_build.cc60 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in InternalKernelBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc295 std::vector<kernel::KernelTensor *> output_kernel_tensors = in LookupDeviceCache() local
356 std::vector<kernel::KernelTensor *> output_kernel_tensors = in UpdateDeviceCache() local
Ddevice_embedding_operation.cc292 const std::vector<kernel::KernelTensor *> &output_kernel_tensors, in InferOpShape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Ddynamic_shape_helper.cc517 const std::vector<kernel::KernelTensor *> &output_kernel_tensors) { in UpdateKernelTensorShape()
568 …UpdateKernelTensorType(const TypePtr &type, const std::vector<kernel::KernelTensor *> &output_kern… in UpdateKernelTensorType()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/
Daicpu_kernel_build.cc253 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(anf… in AicpuOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc747 std::vector<kernel::KernelTensor *> output_kernel_tensors; in GetOutputKernelTensors() local
884 const auto &output_kernel_tensors = GetOutputKernelTensors(output_edges, device_context); in RunSingleOpGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/hardware/
Dcpu_device_context.cc467 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(nod… in CreateKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_kernel_executor.cc103 …std::vector<KernelTensor *> output_kernel_tensors = AnfAlgo::GetOrCreateAllOutputKernelTensors(ker… in CreateKernelPacketKernelMod() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Danf_runtime_algorithm.cc1041 std::vector<KernelTensor *> output_kernel_tensors(output_num); in GetOrCreateAllOutputKernelTensors() local