Searched defs:dev_shape (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/ |
| D | gpu_kernel_utils.h | 34 size_t *dev_shape, size_t *dev_axis, T *dst, cudaStream_t cuda_stream, in MatrixTransposeND()
50 … const std::vector<size_t> host_axis, size_t *dev_shape, size_t *dev_axis, cuComplex *dst, in MatrixTransposeND()
58 … const std::vector<size_t> host_axis, size_t *dev_shape, size_t *dev_axis, in MatrixTransposeND()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | device_matrix.cc | 34 DeviceMatrix::DeviceMatrix(int64_t rank, RankList dev_list, Shape dev_shape) in DeviceMatrix()
130 Shape ConvertRankToCoordinate(int64_t rank, const Shape &dev_shape) { in ConvertRankToCoordinate()
|
| D | parameter_manager.cc | 1052 auto dev_shape = target_param_layout->device_arrangement().array(); in GetRankListByLayout() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | arithmetic_info.cc | 109 Shape dev_shape; in InferDevMatrixShape() local
318 Shape dev_shape = dev_matrix_shape_; in InferTensorMap() local
561 Shape dev_shape = dev_matrix_shape_; in InferTensorMap() local
|
| D | quant_info.cc | 531 Shape dev_shape = dev_matrix_shape_; in InferTensorMap() local
|
| D | reduce_method_info.cc | 560 Shape dev_shape; in InferDevMatrixShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/ |
| D | acl_kernel_mod.cc | 117 ShapeVector dev_shape; in PackageOutput() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/ |
| D | op_api_convert.h | 384 const auto &dev_shape = get_dev_shape(tensor_format, tensor_shape); in GetViewShapeAndStride() local
394 const auto &dev_shape = get_dev_shape(tensor_format, storage_info->ori_shape); in GetViewShapeAndStride() local
|
| D | acl_convert.h | 56 std::string dev_shape; member
|
| D | acl_convert.cc | 133 …pAclString(const aclDataType data_type, const ShapeVector &ori_shape, const ShapeVector &dev_shape, in DumpAclString()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | transform_util.cc | 169 const ShapeVector &dev_shape, in GetGeTensorDesc()
|
| D | convert.cc | 4522 std::shared_ptr<GeTensorDesc> GeOpConvertor::GetTensorDesc(const ShapeVector &dev_shape, const Type… in GetTensorDesc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/ |
| D | ms_device_shape_transfer.cc | 430 auto dev_shape = GetFixedDeviceShape(shape, node, index, is_output); in GetDeviceShapeByFormat() local
|