| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/ |
| D | mul_int8_tests.cc | 37 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
111 std::vector<int> shape2 = {2, 1, 3}; in TEST_F() local
185 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
259 std::vector<int> shape2 = {2, 1, 2}; in TEST_F() local
333 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/lite/test/ut/src/runtime/kernel/arm/fp32/ |
| D | sparse_to_dense_fp32_tests.cc | 37 std::vector<int> shape2 = {2}; in TEST_F() local
124 std::vector<int> shape2 = {2}; in TEST_F() local
211 std::vector<int> shape2 = {2}; in TEST_F() local
296 std::vector<int> shape2 = {1}; in TEST_F() local
381 std::vector<int> shape2 = {2}; in TEST_F() local
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| 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_clip_by_norm_no_div_sum.py | 47 def test_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/tests/ut/cpp/ops/ |
| D | test_ops_unstack.cc | 54 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_custom_predict.cc | 55 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_unpack.cc | 54 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_hashtable_lookup.cc | 56 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_layer_norm_beta_gamma_backprop_v2.cc | 58 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_topk.cc | 67 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_merge.cc | 57 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_layer_norm_x_backprop_v2.cc | 63 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_detection_post_process.cc | 55 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_batch_norm.cc | 56 auto shape2 = shape_vec[1]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| D | test_ops_batch_norm_grad.cc | 62 auto shape2 = shape_vec[2]->cast<abstract::ShapePtr>()->shape(); in TEST_F() local
|
| /third_party/mindspore/mindspore/ops/composite/ |
| D | math_ops.py | 671 def _infer_shape_rem(shape1, shape2, ndim1, ndim2, transpose_b): argument
686 def _check_matmul_shapes(shape1, shape2, prim_name=None): argument
718 def _check_need_broadcast(shape1, shape2): argument
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_utils.h | 88 bool SameDims(const std::vector<T1> &shape1, const std::vector<T2> &shape2) { in SameDims()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | add_relu_grad_v2_fusion.cc | 80 std::vector<size_t> shape2 = AnfAlgo::GetPrevNodeOutputInferShape(tensor_add, 1); in Process() local
|
| D | add_relu_v2_fusion.cc | 78 std::vector<size_t> shape2 = AnfAlgo::GetPrevNodeOutputInferShape(tensor_add, 1); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/ |
| D | broadcast_complex_gpu_kernel.h | 62 auto shape2 = AnfAlgo::GetInputRealDeviceShapeIfExist(kernel_node, 1); in Init() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | utils.cc | 55 ShapePtr CalculateDynamicShape(const ShapePtr &shape1, const ShapePtr &shape2, const ShapeVector &d… in CalculateDynamicShape()
104 ShapePtr ShapeJoin(const ShapePtr &shape1, const ShapePtr &shape2) { in ShapeJoin()
|