| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/ |
| D | infer_device_address.cc | 31 bool InferDeviceAddress::SyncDeviceToHost(const ShapeVector &, size_t size, TypeId type, void *host… in SyncDeviceToHost() argument
63 bool InferDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId type, const void… in SyncHostToDevice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_device_address.cc | 141 bool CPUDeviceAddress::SyncDeviceToHost(const ShapeVector &, size_t size, TypeId type, void *host_p… in SyncDeviceToHost() argument
186 bool CPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId type, const void *… in SyncHostToDevice()
248 bool CPUDeviceAddress::SyncDeviceToDevice(const ShapeVector &, size_t size, TypeId type, const void… in SyncDeviceToDevice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/mindapi/base/ |
| D | shape_vector.h | 24 using ShapeVector = std::vector<ShapeValueDType>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/acl_declare/ |
| D | batch_normalization.cc | 22 std::string Set5HDFormat(TypeId, const std::vector<ShapeVector> &) { return kOpFormat_NC1HWC0; } in Set5HDFormat()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/ |
| D | gpu_device_address.cc | 133 bool GPUDeviceAddress::SyncDeviceToHost(const ShapeVector &, size_t size, TypeId, void *host_ptr) c… in SyncDeviceToHost() argument
164 bool GPUDeviceAddress::SyncHostToDevice(const ShapeVector &, size_t size, TypeId, const void *host_… in SyncHostToDevice()
204 bool GPUDeviceAddress::SyncDeviceToDevice(const ShapeVector &, size_t size, TypeId type, const void… in SyncDeviceToDevice()
294 bool GPUDeviceAddress::AsyncHostToDevice(const ShapeVector &, size_t size, TypeId, const void *host… in AsyncHostToDevice()
307 bool GPUDeviceAddress::AsyncDeviceToHost(const ShapeVector &, size_t size, TypeId, void *host_ptr, in AsyncDeviceToHost()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/ |
| D | graph_split_info.h | 25 using ShapeVector = std::vector<int64_t>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/ |
| D | device_address.h | 183 …virtual bool AsyncHostToDevice(const ShapeVector &, size_t, TypeId, const void *, size_t) const { … in AsyncHostToDevice()
185 …virtual bool AsyncDeviceToHost(const ShapeVector &, size_t, TypeId, void *, size_t) const { return… in AsyncDeviceToHost()
188 …virtual bool SyncDeviceToDevice(const ShapeVector &, size_t, TypeId, const void *, const std::stri… in SyncDeviceToDevice()
192 …virtual bool AsyncDeviceToDevice(const ShapeVector &, size_t, TypeId, const void *, const std::str… in AsyncDeviceToDevice()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/ |
| D | eigen_tensor.h | 75 using ShapeVector = std::vector<int64_t>; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | ms_device_shape_transfer.h | 63 using ShapeVector = std::vector<int64_t>; variable
|
| D | ms_device_shape_transfer.cc | 455 std::optional<ShapeVector> DeviceShapeTransfer::GetFixedDeviceShape(const ShapeVector &, const AnfN… in GetFixedDeviceShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | ascend_device_address.cc | 726 bool AscendDeviceAddress::AsyncDeviceToDevice(const ShapeVector & /* shape */, size_t size, TypeId … in AsyncDeviceToDevice()
797 bool AscendDeviceAddress::AsyncHostToDevice(const ShapeVector & /* shape */, size_t size, TypeId /*… in AsyncHostToDevice()
819 bool AscendDeviceAddress::AsyncDeviceToHost(const ShapeVector & /* shape */, size_t size, TypeId /*… in AsyncDeviceToHost()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | deformable_offsets.cc | 54 using ShapeVector = std::vector<int64_t>; typedef
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/format_transfer/ |
| D | format_transfer_fractal_nz.cc | 47 using ShapeVector = std::vector<int64_t>; typedef
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | dense_to_sparse_set_operation_cpu_kernel.cc | 277 … const std::map<ShapeVector, std::set<T>> &sets) { in OutputSparseTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/ |
| D | config_file_parser.cc | 50 using ShapeVector = std::vector<int64_t>; typedef
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | neighbor_exchange_v2_unify_mindir.cc | 823 … const std::pair<ShapeVector, BaseShapePtr> &shape_info, in CreatePadNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 215 std::string ToString(const TypeId, const ShapeVector &, bool use_comma) const override { in ToString() argument
|