| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.h | 403 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
441 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
444 …double GetBackwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64… in GetBackwardCommCost()
451 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost()
455 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost()
474 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
477 …double GetBackwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64… in GetBackwardCommCost()
485 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost()
490 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost()
634 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
[all …]
|
| D | operator_costmodel.cc | 58 double OperatorCost::GetOutputMemoryCost(const std::vector<TensorInfo> &, in GetOutputMemoryCost()
70 double OperatorCost::GetMemoryCostForInference(const std::vector<TensorInfo> &, in GetMemoryCostForInference()
101 double MatMulCost::GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<Ten… in GetBackwardCommCost()
148 double MatMulCost::GetBackwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vec… in GetBackwardComputationCost()
202 double BatchNormCost::GetForwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<T… in GetForwardCommCost()
220 double BatchNormCost::GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<… in GetBackwardCommCost()
252 double BatchNormCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::v… in GetForwardComputationCost()
263 double BatchNormCost::GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector… in GetBackwardComputationCost()
279 double CastCost::GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> … in GetForwardCommCost() argument
285 double CastCost::GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<Tenso… in GetBackwardCommCost()
[all …]
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | group_convolution_creator.h | 28 struct TensorInfo { struct
29 std::vector<int> shape_;
30 AllocatorPtr allocator_;
31 mindspore::Format format_;
32 TypeId data_type_;
33 lite::Category tensor_type_;
34 bool is_in_;
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/model_pool/ |
| D | model_pool.h | 36 struct TensorInfo { struct
37 std::string name = "";
38 enum DataType data_type;
39 std::vector<int64_t> shape;
40 mindspore::Format format;
41 std::vector<QuantParam> quant_param;
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/ |
| D | helper_base.h | 37 struct TensorInfo { struct
38 std::vector<std::vector<int>> shapes;
39 std::vector<std::vector<TypeId>> types;
63 virtual TensorInfo GetOutputTensorInfo() { return TensorInfo(); } in GetOutputTensorInfo() argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | fullconnection_int8_tests.cc | 32 struct TensorInfo { struct
33 float *data;
34 int *data_int;
35 float min;
36 float max;
37 int len;
38 std::vector<int> *shape;
|
| D | matmul_int8_tests.cc | 33 struct TensorInfo { struct
34 float *data;
35 float min;
36 float max;
37 int len;
38 std::vector<int> *shape;
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_info.h | 34 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo() function
36 explicit TensorInfo(const TensorLayout &tensor_layout) : tensor_layout_(tensor_layout) { in TensorInfo() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | kernel.h | 230 struct TensorInfo { struct
243 using TensorInfoPtr = std::shared_ptr<TensorInfo>; argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/ |
| D | tensor_info.cc | 54 TensorInfo::TensorInfo(const std::string &name, mindspore::DataType type, const std::vector<int64_t… in TensorInfo() function in mindspore::lite::TensorInfo
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/ |
| D | mem_copy_manager.h | 37 struct TensorInfo { struct
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/offline_debug/ |
| D | dbg_services.py | 384 class TensorInfo: class
|