/third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
D | operator_costmodel.h | 125 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost() 135 …e GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetComputationCost() 153 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost() 161 …e GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetComputationCost() 267 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost() 275 …e GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetComputationCost() 324 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost() 332 …e GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetComputationCost() 352 double GetCommCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetCommCost() 360 …e GetComputationCost(const std::vector<TensorInfo> &inputs, const std::vector<TensorInfo> &outputs, in GetComputationCost() [all …]
|
/third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/ |
D | depthwise_conv2d_infer_test.cc | 41 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 87 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 133 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 180 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 227 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 274 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 321 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 368 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 415 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 462 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local [all …]
|
D | conv2d_infer_test.cc | 41 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 86 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 131 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 177 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 223 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 269 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 315 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 361 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 407 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 453 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local [all …]
|
D | reshape_infer_test.cc | 34 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 66 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 98 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 130 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 161 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 193 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 227 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 261 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 294 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 329 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | strided_slice_infer_test.cc | 34 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 76 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 118 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 158 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 190 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 245 std::vector<TensorC *> outputs; in TEST_F() local 293 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | concat_infer_test.cc | 37 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 67 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 99 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 134 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 170 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 218 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | unsqueeze_infer_test.cc | 33 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 59 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 85 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 112 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 144 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 176 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | pooling_infer_test.cc | 35 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 75 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 116 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 157 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 198 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 239 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | depth_to_space_infer_test.cc | 37 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 68 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 99 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 130 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 160 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local
|
D | reduce_infer_test.cc | 33 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 61 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 91 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 124 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 158 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | mean_infer_test.cc | 34 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 61 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 90 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 122 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 155 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | gather_nd_infer_test.cc | 38 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 69 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 100 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 130 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 163 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | gather_infer_test.cc | 38 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 70 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 102 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 133 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 166 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | split_infer_test.cc | 33 std::vector<TensorC *> outputs(3, NULL); in TEST_F() local 72 std::vector<TensorC *> outputs(2, NULL); in TEST_F() local 109 std::vector<TensorC *> outputs(3, NULL); in TEST_F() local 158 std::vector<TensorC *> outputs(2, NULL); in TEST_F() local 197 std::vector<TensorC *> outputs(100, NULL); in TEST_F() local
|
D | pad_infer_test.cc | 39 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 64 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 95 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 135 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 169 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | fill_infer_test.cc | 36 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 65 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 92 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local 117 std::vector<TensorC *> outputs(inputs_size, NULL); in TEST_F() local
|
D | argmin_infer_test.cc | 32 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 59 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 87 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 115 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | argmax_infer_test.cc | 32 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 59 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 87 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 115 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
D | space_to_batch_nd_infer_test.cc | 37 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 74 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 111 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local 148 std::vector<TensorC *> outputs(1, NULL); in TEST_F() local
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/control/ |
D | tensor_array.h | 30 const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in TensorArrayCPUKernel() 49 … const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in TensorArrayBaseCPUKernel() 65 … const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in TensorArrayReadCPUKernel() 77 … const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in TensorArrayWriteCPUKernel()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
D | isnan_cpu_kernel.cc | 39 const std::vector<kernel::AddressPtr> &outputs) { in Launch() 55 const std::vector<kernel::AddressPtr> &outputs) { in LaunchKernelFloat16() 69 const std::vector<kernel::AddressPtr> &outputs) { in LaunchKernelFloat() 81 const std::vector<kernel::AddressPtr> &outputs) { in LaunchKernelOther()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
D | compose_op.cc | 25 …ComposeOp::OutputShape(const std::vector<TensorShape> &inputs, std::vector<TensorShape> &outputs) { in OutputShape() 34 Status ComposeOp::OutputType(const std::vector<DataType> &inputs, std::vector<DataType> &outputs) { in OutputType() 43 Status ComposeOp::Compute(const TensorRow &inputs, TensorRow *outputs) { in Compute()
|
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
D | convolution_int8_creator.cc | 42 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvDwInt8KernelCreator() 77 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelSelect() 102 const std::vector<lite::Tensor *> &outputs, in CpuGroupConvInt8KernelCreator() 117 … const std::vector<lite::Tensor *> &outputs, OpParameter *op_parameter, in CpuConvInt8KernelCreator()
|
D | relux_int8.h | 29 const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in ReluXInt8CPUKernel() 49 const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in ReluInt8CPUKernel() 65 const std::vector<lite::Tensor *> &outputs, const lite::InnerContext *ctx) in Relu6Int8CPUKernel()
|
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/ |
D | model_impl.cc | 20 Status ModelImpl::Predict(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) { in Predict() 48 Status ModelImpl::Preprocess(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) { in Preprocess() 79 …lImpl::PredictWithPreprocess(const std::vector<MSTensor> &inputs, std::vector<MSTensor> *outputs) { in PredictWithPreprocess()
|