| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | matmul_fp32_tests.cc | 74 …onst std::vector<int> &a_shape, const std::vector<int> &b_shape, const std::vector<int> &c_shape) { in MMTestInit()
94 const std::vector<int> &bias_shape, const std::vector<int> &c_shape) { in MMTestInit2()
134 std::vector<int> c_shape = {2, 3}; in TEST_F() local
172 std::vector<int> c_shape = {2, 3}; in TEST_F() local
228 std::vector<int> c_shape = {25, 36}; in TEST_F() local
302 std::vector<int> c_shape = {1, 2, 3}; in TEST_F() local
356 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-src/source/mindspore/core/ops/ops_func_impl/ |
| D | apply_came_part4.cc | 88 …auto c_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex3]->GetShape(… in CheckInputsShapePart4() local
147 ShapeVector c_shape = out_shape; in InferShape() local
|
| D | apply_came_part2.cc | 97 …auto c_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex5]->GetShape(… in CheckInputsShapePart2() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | affine_infer.c | 102 int c_shape[MAX_SHAPE_SIZE]; in AffineInferShape() local
|
| D | matmul_infer.c | 110 int c_shape[MAX_SHAPE_SIZE]; in SetShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/java/src/main/native/ |
| D | ms_tensor.cpp | 417 std::vector<int64_t> c_shape(size); in Java_com_mindspore_MSTensor_setShape() local
480 std::vector<int64_t> c_shape(size); in Java_com_mindspore_MSTensor_createTensorByNative() local
571 std::vector<int64_t> c_shape(shape.begin(), shape.end()); in Java_com_mindspore_MSTensor_createTensorByObject() local
|
| D | ms_context.cpp | 154 const char *c_shape = env->GetStringUTFChars(rv, nullptr); in ParseAscendDevInfoInputShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_matrix_mul_cpu_kernel.cc | 84 auto c_shape = reinterpret_cast<T *>(outputs[kOutShapeIdx]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | dynamic_rnn.cc | 117 …auto c_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kDynRnnIdx5]->GetShape()… in DynamicRNNShapeCheck() local
|
| D | lstm_v2.cc | 95 auto c_shape = c_shape_map[kShape]; in LSTMV2InferShape() local
|
| D | lstm.cc | 99 std::vector<int64_t> c_shape = {c_input_shape}; in LSTMInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | nn_ops.py | 3163 def infer_shape(self, x_shape, h_shape, c_shape, w_shape): argument
5972 def infer_shape(self, x_shape, h_shape, c_shape, w_shape, b_shape): argument
|
| D | _grad_ops.py | 1877 …def infer_shape(self, c_shape, dht_shape, dct_shape, it_shape, jt_shape, ft_shape, ot_shape, tanhc… argument
|