| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/internal/ |
| D | internal_kernel_utils.cc | 82 …rnelUtils::ToInternalTensor(internal::Tensor *internal_tensor, const KernelTensor *kernel_tensor) { in ToInternalTensor()
94 internal::DeviceRawBuf InternalKernelUtils::ToDeviceRawBuf(const KernelTensor *kernel_tensor) { in ToDeviceRawBuf()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | device_address_utils.cc | 62 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressForScalarAndString() local
73 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressForScalarAndString() local
79 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressForScalarAndString() local
138 const auto &kernel_tensor = device_address->kernel_tensor(); in CopyNoneTensorDataToDevice() local
183 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressByMapTensorNode() local
246 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateParameterDeviceAddress() local
323 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressForTensorValue() local
384 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateDeviceAddressForTypeValue() local
510 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateKernelOutputDeviceAddress() local
557 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in CreateGraphOutputDeviceAddress() local
[all …]
|
| D | launch_kernel.cc | 32 auto kernel_tensor = std::make_shared<kernel::KernelTensor>(); in ObtainKernelAddress() local
|
| D | move_to.cc | 110 auto kernel_tensor = std::make_shared<kernel::KernelTensor>( in MoveTo() local
|
| D | kernel_info.cc | 32 bool KernelInfo::SetOutputKernelTensor(const KernelTensorPtr &kernel_tensor, size_t index) { in SetOutputKernelTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | ops_utils.cc | 87 KernelTensor *kernel_tensor) { in ComputeBeginMask()
99 KernelTensor *kernel_tensor) { in ComputeEndMask()
111 const ShapeVector &input_shape, KernelTensor *kernel_tensor) { in ComputeEllipsisMask()
127 const ShapeVector &input_shape, KernelTensor *kernel_tensor) { in ComputNewAxisMask()
143 KernelTensor *kernel_tensor) { in ComputeShrinkAxisMask()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ |
| D | device_embedding_operation.cc | 183 auto kernel_tensor = std::make_shared<kernel::KernelTensor>( in MemcpyHostToDeviceAsync() local
205 auto kernel_tensor = std::make_shared<kernel::KernelTensor>( in MemcpyDeviceToHostAsync() local
271 const auto &kernel_tensor = AnfAlgo::CreateOutputKernelTensorWithDeviceInfo( in NewValueNode() local
|
| D | device_dense_embedding_operation.cc | 310 …const auto &kernel_tensor = AnfAlgo::GetPrevNodeOutputKernelTensor(embedding_cache_lookup_node_, i… in LookupDeviceCache() local
371 …const auto &kernel_tensor = AnfAlgo::GetPrevNodeOutputKernelTensor(embedding_cache_update_node_, i… in UpdateDeviceCache() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/session/ |
| D | single_op_session.cc | 149 auto kernel_tensor = new (std::nothrow) kernel::KernelTensor(); in BuildCustomAscendKernelImpl() local
173 auto kernel_tensor = make_kernel_tensor(static_cast<TypeId>(data_type), shape); in BuildCustomAscendKernelImpl() local
184 auto kernel_tensor = new (std::nothrow) kernel::KernelTensor(); in BuildCustomAscendKernelImpl() local
246 auto &kernel_tensor = kernel_args_.inputs[i]; in InitInputOutputInfos() local
261 auto &kernel_tensor = kernel_args_.outputs[i]; in InitInputOutputInfos() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_ext_info_handle.cc | 208 …oHandler::UpdateInputShapeAndType(uint32_t input_index, const kernel::KernelTensor *kernel_tensor)… in UpdateInputShapeAndType()
222 …andler::UpdateOutputShapeAndType(uint32_t output_index, const kernel::KernelTensor *kernel_tensor)… in UpdateOutputShapeAndType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/ |
| D | hccl_kernel.cc | 198 GetTensorShape = [](KernelTensor *kernel_tensor) { in CalcTypeShapeAndCount()
205 GetTensorShape = [](KernelTensor *kernel_tensor) { return kernel_tensor->GetShapeVector(); }; in CalcTypeShapeAndCount()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/rpc/ |
| D | rpc_recv_kernel.cc | 83 … [](const auto &kernel_tensor) { return kernel_tensor->IsDynamicShape(); }); in Init()
|
| D | rpc_send_kernel.cc | 28 … [](const auto &kernel_tensor) { return kernel_tensor->IsDynamicShape(); }); in Init()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ |
| D | ge_device_res_manager.h | 43 explicit GeHostAddress(const KernelTensorPtr &kernel_tensor) : CPUDeviceAddress(kernel_tensor) {} in GeHostAddress()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/common/ |
| D | device_common_test.h | 56 TestDeviceAddress(const KernelTensorPtr &kernel_tensor) : DeviceAddress(kernel_tensor) {} in TestDeviceAddress()
101 DeviceAddressPtr CreateDeviceAddress(const KernelTensorPtr &kernel_tensor) const { in CreateDeviceAddress()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | kernel_actor.cc | 438 auto kernel_tensor = std::make_shared<kernel::KernelTensor>( in FetchWorkspaceDeviceTensor() local
535 const auto &kernel_tensor = output_kernel_tensors_[i]; in TraceDynamicMemory() local
544 const auto &kernel_tensor = workspace_kernel_tensors_[i]; in TraceDynamicMemory() local
779 const auto &kernel_tensor = input_device_tensors_[input_index]->kernel_tensor(); in UpdateInputDeviceTensor() local
|
| D | abstract_actor.cc | 138 …const auto &kernel_tensor = (*input_device_tensors)[device_tensor_store_key.first]->kernel_tensor(… in FetchInputByTensorStore() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_device_address.h | 30 explicit CPUDeviceAddress(const KernelTensorPtr &kernel_tensor) : DeviceAddress(kernel_tensor) { in CPUDeviceAddress()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_device_address.h | 35 …explicit GPUDeviceAddress(const KernelTensorPtr &kernel_tensor) : LoadableDeviceAddress(kernel_ten… in GPUDeviceAddress()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/ |
| D | dynamic_shape_helper.cc | 529 const auto &kernel_tensor = output_kernel_tensors[i]; in UpdateKernelTensorShape() local
534 const auto &kernel_tensor = output_kernel_tensors[0]; in UpdateKernelTensorShape() local
574 const auto &kernel_tensor = output_kernel_tensors[0]; in UpdateKernelTensorType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runner.cc | 416 auto kernel_tensor = std::make_shared<KernelTensor>( in GetWorkspaceKernelTensors() local
466 auto kernel_tensor = std::make_shared<KernelTensor>( in GetWorkspaceKernelTensorsDynamic() local
568 const auto &kernel_tensor = device_address->kernel_tensor(); in UpdateOutputShape() local
668 const auto &kernel_tensor = device_address->kernel_tensor(); in UpdateOutputDeviceInfo() local
710 const auto &kernel_tensor = origin_address->kernel_tensor(); in UpdateAddressInfoByInputTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/ |
| D | mem_tracker.cc | 122 … KernelTensorPtr kernel_tensor, const std::string &file_name, in NewMemInfo()
147 … KernelTensorPtr kernel_tensor, const std::string &file_name, in AddMemInfoForKernelTensor()
177 auto kernel_tensor = device_address->kernel_tensor().get(); in UpdateMemInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | ascend_device_address.h | 42 …explicit AscendDeviceAddress(const KernelTensorPtr &kernel_tensor) : LoadableDeviceAddress(kernel_… in AscendDeviceAddress()
|
| D | launch_transdata.cc | 113 auto kernel_tensor = std::make_shared<kernel::KernelTensor>( in CreateOutputAddr() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/kernel_packet/ |
| D | kernel_packet_kernel_mod.cc | 70 auto kernel_tensor = std::make_shared<KernelTensor>(abs->GetShape(), abs->GetType(), value); in InitKernel() local
|