| /third_party/mindspore/mindspore/lite/src/delegate/npu/op/ |
| D | convolution_base_npu.cc | 54 auto w_shape = inputs[1].Shape(); in InitWeightConst() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | prelu.cc | 28 auto w_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(w)[kShape]; in InferShape() local
|
| D | conv2d.cc | 155 auto w_shape = w_shape_map[kShape]; in Conv2dInferShape() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | gru_fp16.cc | 70 std::vector<int> w_shape = weight_g->shape(); in InitParam() local
|
| D | lstm_fp16.cc | 75 std::vector<int> w_shape = weight_i->shape(); in InitParam() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | gru_fp32.cc | 71 std::vector<int> w_shape = weight_g->shape(); in InitParam() local
|
| D | lstm_fp32.cc | 170 std::vector<int> w_shape = weight_i->shape(); in InitParam() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | lstm_fp32_coder.cc | 100 std::vector<int> w_shape = weight_i->shape(); in InitParam() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_nn.cc | 234 void CheckShape(const std::string &op_name, const ShapeVector &w_shape, const AbstractTensorPtr &in… in CheckShape()
261 ShapeVector w_shape = input_w->shape()->shape(); in InferImplConv2D() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 1248 def infer_shape(self, y_shape, dy_shape, dhy_shape, dcy_shape, w_shape, argument
1330 …def infer_shape(self, x_shape, hx_shape, cx_shape, w_shape, y_shape, hy_shape, cy_shape, dy_shape,… argument
1382 def infer_shape(self, x_shape, w_shape, b_shape, y_shape, init_h_shape, init_c_shape, h_shape, argument
1569 def infer_shape(self, y_backprop_shape, a_shape, w_shape): argument
1818 def infer_shape(self, x_shape, y_grad_shape, t_shape, w_shape, tw_shape): argument
2149 def infer_shape(self, dgate_shape, w_shape): argument
|
| D | nn_ops.py | 1562 def infer_shape(self, x_shape, w_shape, b_shape=None): argument
2455 def infer_shape(self, x_shape, t_shape, w_shape): argument
3971 def infer_shape(self, x_shape, h_shape, c_shape, w_shape): argument
7588 def infer_shape(self, x_shape, h_shape, c_shape, w_shape, b_shape): argument
7745 def infer_shape(self, x_shape, w_shape, b_shape, seq_shape, h_shape, c_shape): argument
8356 def infer_shape(self, x_shape, w_shape, b_shape=None): argument
|
| D | _quant_ops.py | 1703 def infer_shape(self, w_shape, w_min_shape, w_max_shape): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_op_converter.cc | 541 const auto &w_shape = AnfAlgo::GetPrevNodeOutputInferShape(node, 1); in MS_TRT_CONVERTER_FUNC_REG() local
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1410 auto w_shape = dyn_cast<abstract::Shape>(node->input(kTwoNum)->Shape()); in ExportPrimDepthwiseConv2d() local
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_ops.py | 575 …def __init__(self, w_shape, out_channel, kernel_size, mode, pad_mode, pad, stride, dilation, group… argument
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 672 int64_t w_shape = shape[shape.size() - kDim1]; in FracNZDeviceDynamicShape() local
|