| /third_party/mindspore/mindspore-src/source/tests/st/ops/ |
| D | test_ops_matmul_ext.py | 62 def test_ops(context_mode, shape1, shape2, output_shape1, output_shape2): argument
92 def test_ops_f16(context_mode, shape1, shape2): argument
117 def test_ops_bf16(context_mode, shape1, shape2): argument
|
| D | test_ops_batch_matmul_ext.py | 46 def test_ops(context_mode, shape1, shape2): argument
77 def test_ops_network(context_mode, shape1, shape2, dtype): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/symbol_engine/ops/ |
| D | test_elemwise_binop.cc | 26 ShapeVector shape2; member
44 auto shape2 = b->abstract()->GetSymbolicShape(); in TEST_P() local
|
| D | test_matmul.cc | 27 ShapeVector shape2; member
67 auto shape2 = b->abstract()->GetSymbolicShape(); in TEST_P() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | mul_int8_tests.cc | 38 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
112 std::vector<int> shape2 = {2, 1, 3}; in TEST_F() local
186 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
260 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
334 std::vector<int> shape2 = {2, 3}; in TEST_F() local
|
| D | concat_int8_tests.cc | 37 std::vector<int> shape2 = {3, 2}; in TEST_F() local
112 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
188 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/ |
| D | test_dvm_profiler.py | 30 def fuse(shape1, shape2, dtype): argument
|
| D | test_dvm.py | 71 def fuse(shape1, shape2, dtype): argument
95 def test_easy_fuse_dvm(shape1, shape2, dtype): argument
|
| D | test_squared_difference.py | 39 def test_squared_difference(shape1, shape2, dtype): argument
|
| D | test_relu_grad.py | 39 def test_relu_grad(shape1, shape2, dtype): argument
|
| D | test_bias_add.py | 39 def test_bias_add(shape1, shape2, data_format, dtype): argument
|
| D | test_gelu_grad.py | 39 def relu_grad_test(shape1, shape2, dtype): argument
|
| D | test_clip_by_norm_no_div_sum.py | 47 def run_clip_by_norm_no_div_sum(shape0, shape1, shape2, shape3, dtype): argument
|
| D | test_layernorm_stitch.py | 72 def test_layernorm(shape1, shape2, dtype): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | dense_to_sparse_set_operation_cpu_kernel.cc | 76 bool CheckShapesMatch(ShapeVector shape1, ShapeVector shape2) { in CheckShapesMatch()
86 bool GroupShapeFromInputs(ShapeVector shape1, ShapeVector shape2, ShapeVector *const group_shape) { in GroupShapeFromInputs()
186 void ValidateIndices(kernel::KernelTensor *indices, ShapeVector shape2, int64_t set2_nums, int64_t … in ValidateIndices()
340 ShapeVector shape2(shape2_ptr, shape2_ptr + shape2_size); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/core/abstract/ |
| D | utils_test.cc | 58 auto shape2 = std::make_shared<Shape>(shape_vec2); in ShapeJoinCheck() local
73 auto shape2 = std::make_shared<Shape>(shape_vec2); in ShapeJoinCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | matmul_ext.cc | 36 ShapeVector CheckMatMulShapes(const ShapeVector &shape1, const ShapeVector &shape2) { in CheckMatMulShapes()
84 …erShapeRem(const ShapeVector &shape_backbone, const ShapeVector &shape1, const ShapeVector &shape2, in InferShapeRem()
|
| D | addn.cc | 38 bool AddNDynShapeJoin(ShapeVector *shape1, const ShapeVector *shape2) { in AddNDynShapeJoin()
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/mindapi/ |
| D | mindapi_test.cc | 354 ShapeVector shape2{2, 3}; in TEST_F() local
378 ShapeVector shape2{2, 3}; in TEST_F() local
497 ShapeVector shape2{2, 3, 4}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | densetodense_set_operation.cc | 85 uint32_t ChecksShapesMatch(const std::vector<int64_t> &shape1, const std::vector<int64_t> &shape2) { in ChecksShapesMatch()
96 … const std::vector<int64_t> &shape2, std::vector<int64_t> *group_shape) { in GroupShapeFromDenseInputs()
269 const auto shape2 = set2_t->GetTensorShape()->GetDimSizes(); in DoCompute() local
|
| D | densetosparsesetoperation.cc | 63 uint32_t CheckShapesMatch(const std::vector<int64_t> &shape1, const std::vector<int64_t> &shape2) { in CheckShapesMatch()
74 … const std::vector<int64_t> &shape2, std::vector<int64_t> &group_shape) { in GroupsShapeFromInputs()
264 std::vector<int64_t> shape2(set2_shape.size()); in ComputeDenseToSparse() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_reshape_consecutive.py | 52 def __init__(self, shape1, shape2): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/grad_ops/ |
| D | grad_inner_ops.cc | 187 ShapeVector shape2(x_shape.begin() + 2, x_shape.end()); in __anon79cd20190202() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | select_info.cc | 33 …ectInfo::CheckIsBroadcast(const std::vector<int64_t> &shape1, const std::vector<int64_t> &shape2) { in CheckIsBroadcast()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | combined_non_max_suppression_proto.cc | 161 Shape shape2({batch_dim, output_size}); in IMPLEMT_INFERFUNC() local
|