Searched defs:TensorInfo (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| 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()
147 double MatMulCost::GetBackwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vec… in GetBackwardComputationCost()
200 double CastCost::GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> … in GetForwardCommCost() argument
206 double CastCost::GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<Tenso… in GetBackwardCommCost()
228 double CastCost::GetForwardComputationCost(const std::vector<TensorInfo> &inputs, const std::vector… in GetForwardComputationCost()
237 double CastCost::GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<Tens… in GetBackwardComputationCost()
265 double SoftmaxCost::GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInf… in GetForwardCommCost()
272 double SoftmaxCost::GetBackwardCommCost(const std::vector<TensorInfo> &inputs, const std::vector<Te… in GetBackwardCommCost()
[all …]
|
| D | operator_costmodel.h | 356 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
393 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
396 …double GetBackwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64… in GetBackwardCommCost()
403 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost()
407 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost()
426 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
429 …double GetBackwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64… in GetBackwardCommCost()
437 double GetForwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetForwardComputationCost()
442 …double GetBackwardComputationCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, in GetBackwardComputationCost()
586 …double GetForwardCommCost(const std::vector<TensorInfo> &, const std::vector<TensorInfo> &, int64_… in GetForwardCommCost()
[all …]
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | group_convolution_creator.h | 26 struct TensorInfo { struct
27 std::vector<int> shape_;
28 mindspore::Format format_;
29 TypeId data_type_;
30 lite::Tensor::Category tensor_type_;
31 bool is_in_;
|
| /third_party/mindspore/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/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_info.h | 35 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo() function
37 explicit TensorInfo(const TensorLayout &tensor_layout) : tensor_layout_(tensor_layout) { in TensorInfo() function
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_copy_manager.h | 35 struct TensorInfo { struct
|
| /third_party/mindspore/mindspore/offline_debug/ |
| D | dbg_services.py | 370 class TensorInfo: class
|