| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | matmul_tests.cc | 40 std::vector<int> weight_shape = {co, ci}; variable
60 std::vector<int> weight_shape = {a, b, co, ci}; variable
80 std::vector<int> weight_shape = {a, co, ci}; variable
100 std::vector<int> weight_shape = {a, co, ci}; in TEST_F() local
120 std::vector<int> weight_shape = {a, ci, co}; in TEST_F() local
|
| D | fullconnection_tests.cc | 25 OpParameter *CreateParameter(std::vector<int> *input_shape, std::vector<int> *weight_shape, in CreateParameter()
64 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
86 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
109 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
132 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; variable
|
| D | scale_tests.cc | 36 std::vector<int> weight_shape = {input_shape[axis]}; in TEST_F() local
54 std::vector<int> weight_shape = {input_shape[axis]}; in TEST_F() local
72 std::vector<int> weight_shape = {input_shape[axis]}; in TEST_F() local
|
| D | conv2d_transpose_tests.cc | 26 std::vector<int> *input_shape, std::vector<int> *weight_shape, in CreateParameter()
68 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
96 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
124 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
152 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
183 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
213 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
|
| D | batchnorm_tests.cc | 34 std::vector<int> weight_shape = {1, 1, 1, 8}; in TEST_F() local
64 std::vector<int> weight_shape = {1, 1, 1, 7}; in TEST_F() local
|
| D | depthwise_conv2d_tests.cc | 62 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F() local
108 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F() local
155 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F() local
199 std::vector<int> weight_shape = {1, kernel_h, kernel_w, output_shape.back()}; in TEST_F() local
|
| D | conv2d_tests.cc | 44 …std::vector<int> weight_shape = {param->output_channel_, param->kernel_h_, param->kernel_w_, param… in TestMain_Conv2D() local
57 std::vector<int> input_shape, weight_shape, bias_shape, output_shape; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_conv_parser.cc | 65 auto weight_shape = weight_tensor->shape; in Parse() local
122 auto weight_shape = weight_tensor->shape; in Parse() local
|
| D | tflite_conv_transpose_parser.cc | 60 auto weight_shape = weight_tensor->shape; in Parse() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | split_strategy.cc | 27 const std::vector<int64_t> &weight_shape) { in ApproximateFLOPs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | prelu_gpu_kernel.h | 67 auto weight_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); in Init() local
|
| D | prelu_grad_gpu_kernel.h | 70 auto weight_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2); in Init() local
|
| D | bce_with_logits_loss_kernel.h | 47 size_t *weight_shape = GetDeviceAddress<size_t>(workspace, 1); in Launch() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | lerp.cc | 41 auto weight_shape = weight_shape_map[kShape]; in InferShape() local
|
| D | binary_cross_entropy.cc | 39 …auto weight_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in BinaryCrossEntroyInferShape() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/ |
| D | onnx_conv_transpose_parser.cc | 95 std::vector<int> weight_shape; in Parse() local
|
| D | onnx_conv_parser.cc | 39 std::vector<int> weight_shape; in GetConvChannel() local
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | binary_cross_entropy_grad.cc | 32 …auto weight_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->Buil… in BinaryCrossEntroyGradInferShape() local
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | group_depthwise_op_convert_pass.cc | 79 auto weight_shape = weight_shape_ptr->shape(); in Run() local
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | matmul_int8_tests.cc | 104 std::vector<int> weight_shape{1, 3, 8}; in TEST_F() local
215 std::vector<int> weight_shape{1, 17, 10}; in TEST_F() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/ |
| D | prelu.cc | 134 cl_int4 weight_shape = {}; in Prepare() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/fl/ |
| D | get_model_kernel.h | 125 auto weight_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, i); in Init() local
|
| D | fused_push_weight_kernel.h | 117 auto weight_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, i); in Init() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/ |
| D | conv2d_grad_filter_cpu_kernel.cc | 37 std::vector<size_t> weight_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); in InitKernel() local
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | parser_utils.cc | 254 ShapeVector weight_shape; in UnifyVariableConvWeight() local
360 auto weight_shape = weight_value->shape(); in HandleConstConvWeightShared() local
|