| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | power_fp32_tests.cc | 30 const std::vector<int> &c_shape) { in PowerTestInit()
49 const std::vector<int> &a_shape, const std::vector<int> &c_shape) { in PowerTestInit2()
72 std::vector<int> c_shape = {2, 2}; in TEST_F() local
97 std::vector<int> c_shape = {2, 2}; in TEST_F() local
|
| D | matmul_fp32_tests.cc | 72 …onst std::vector<int> &a_shape, const std::vector<int> &b_shape, const std::vector<int> &c_shape) { in MMTestInit()
92 const std::vector<int> &bias_shape, const std::vector<int> &c_shape) { in MMTestInit2()
130 std::vector<int> c_shape = {2, 3}; in TEST_F() local
164 std::vector<int> c_shape = {2, 3}; in TEST_F() local
217 std::vector<int> c_shape = {25, 36}; in TEST_F() local
288 std::vector<int> c_shape = {1, 2, 3}; in TEST_F() local
339 std::vector<int> c_shape = {3, 2, 3}; in TEST_F() local
|
| D | roi_pooling_fp32_tests.cc | 30 const std::vector<int> &c_shape) { in ROIPoolingTestInit()
60 std::vector<int> c_shape = {1, 2, 2, 2}; in TEST_F() local
|
| D | constant_of_shape_fp32_test.cc | 35 std::vector<int> c_shape(in_t->ElementsNum()); in ConstantOfShapeTestInit() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | matmul_infer.c | 97 int c_shape[MAX_SHAPE_SIZE]; in MatmulInferShape() local
|
| D | affine_infer.c | 102 int c_shape[MAX_SHAPE_SIZE]; in AffineInferShape() local
|
| /third_party/boost/libs/python/test/numpy/ |
| D | ndarray.cpp | 22 Py_intptr_t *c_shape = new Py_intptr_t[len]; in c_empty() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | lstm.cc | 86 std::vector<int64_t> c_shape = {c_input_shape}; in LstmInfer() local
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 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
|
| D | _grad_ops.py | 1382 def infer_shape(self, x_shape, w_shape, b_shape, y_shape, init_h_shape, init_c_shape, h_shape, argument
2069 …def infer_shape(self, c_shape, dht_shape, dct_shape, it_shape, jt_shape, ft_shape, ot_shape, tanhc… argument
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1777 int64_t c_shape = 1; in ExportMergeLayerNorm() local
|