| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/ |
| D | fake_abstract_shape.cc | 31 …static ShapeVector GetFakeAbstractShape(const ShapeVector &device_shape, const std::string &format… in GetFakeAbstractShape()
62 static ShapeVector NchwAbstractShape(const ShapeVector &device_shape) { return device_shape; } in NchwAbstractShape()
63 static ShapeVector NhwcAbstractShape(const ShapeVector &device_shape) { in NhwcAbstractShape()
74 static ShapeVector FractalNzAbstractShape(const ShapeVector &device_shape) { in FractalNzAbstractShape()
99 static ShapeVector Nc1hwc0AbstractShape(const ShapeVector &device_shape) { in Nc1hwc0AbstractShape()
122 ShapeVector GetFakeAbstractShape(const ShapeVector &device_shape, const std::string &format) { in GetFakeAbstractShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/ |
| D | op_runtime_info.cc | 28 … const std::string &format, const ShapeVector &device_shape) { in OpRuntimeInfoGetOutputTensorMemSize()
68 auto device_shape = AnfAlgo::GetOutputDeviceShape(node, i); in CacheForExecutionOrder() local
116 auto device_shape = AnfAlgo::GetOutputDeviceShape(input, index); in CacheForGraphInputs() local
228 auto device_shape = AnfAlgo::GetOutputDeviceShape(node, i); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | quant_dtype_cast_adjust.cc | 58 … auto device_shape = trans::TransShapeToDevice(host_shape, kOpFormat_FRAC_NZ, kNumberTypeFloat16); in Process() local
|
| D | centralization_mindir.cc | 37 auto device_shape = trans::TransShapeToDevice(shape, kOpFormat_FRAC_Z, type); in CheckDeviceShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/dump/ |
| D | ascend_dump.h | 45 ShapeVector device_shape; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | ms_device_shape_transfer.cc | 531 ShapeVector device_shape; in NHWCDeviceShape() local
543 ShapeVector device_shape; in HWCNDeviceShape() local
555 ShapeVector device_shape; in FRAC_ZDeviceShape() local
578 ShapeVector device_shape; in NC1HWC0DeviceShape() local
596 ShapeVector device_shape; in NDC1HWC0DeviceShape() local
612 ShapeVector device_shape; in FRAC_Z3DDeviceShape() local
632 ShapeVector device_shape; in C1HWNCOC0DeviceShape() local
651 ShapeVector device_shape; in FRAC_ZC04DeviceShape() local
672 ShapeVector device_shape; in NC1HWC04DeviceShape() local
698 ShapeVector device_shape; in ChannelLastDeviceShape() local
[all …]
|
| D | ms_device_shape_transfer.h | 85 ShapeVector device_shape; member
|
| D | device_address_utils.cc | 573 auto device_shape = shape; in GetTensorDeviceSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/other/ |
| D | dynamic_stitch_gpu_kernel.cc | 62 auto device_shape = data[index]->GetShapeVector(); in GetShapeAdaptively() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/ |
| D | fse_decode_kernel.cc | 69 int NCHW_TO_FRAC_NZ(const std::vector<int> &host_shape, const std::vector<int> &device_shape, in NCHW_TO_FRAC_NZ()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | inplace_assign_builder.cc | 157 auto device_shape = GetDeviceShape(op_info.op_node); in CreateCleanCompositeNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/ |
| D | ascend_device_address.cc | 460 ShapeVector device_shape; in GetDeviceShape() local
548 auto device_shape = GetDeviceShape(&host_shape); in SyncDeviceToHostAndConvertFormat() local
852 std::vector<int64_t> device_shape; in ConvertFormatAndSyncHostToDevice() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/ |
| D | dynamic_utils.cc | 538 std::vector<std::vector<int64_t>> device_shape(device_shape_list_); in GetDeviceArgSizeVec() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/ |
| D | device_address.h | 244 const ShapeVector &device_shape() const { return device_shape_; } in device_shape() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/ |
| D | data_prepare_actor.cc | 348 auto device_shape = in UpdateDataNodeDeviceAddressSize() local
|