Home
last modified time | relevance | path

Searched refs:shp_x (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/tests/ut/cpp/pre_activate/pass/
Dconvert_const_input_to_attr_test.cc43 std::vector<int64_t> shp_x{2, 3}; in TEST_F() local
44 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
57 std::vector<int64_t> shp_x{2, 3}; in TEST_F() local
58 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
71 std::vector<int64_t> shp_x{2, 2, 3}; in TEST_F() local
72 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
94 std::vector<int64_t> shp_x{16}; in TEST_F() local
95 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kInt32, shp_x); in TEST_F()
Dconvert_const_input_to_tensor_input_test.cc43 std::vector<int64_t> shp_x{16}; in TEST_F() local
44 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kInt32, shp_x); in TEST_F()
72 std::vector<int64_t> shp_x{16}; in TEST_F() local
73 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kInt32, shp_x); in TEST_F()
90 std::vector<int64_t> shp_x{1}; in TEST_F() local
91 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dallreduce_fusion_test.cc46 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
47 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
87 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
88 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
128 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
129 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
179 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
180 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dcommon_subexpression_elimination_test.cc58 std::vector<int64_t> shp_x{32, 3, 224, 224}; in TEST_F() local
59 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
86 std::vector<int64_t> shp_x{32, 3, 224, 224}; in TEST_F() local
87 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
129 std::vector<int64_t> shp_x{32, 3, 224, 224}; in TEST_F() local
130 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dconvert_tuple_input_to_dynamic_input_test.cc42 std::vector<int64_t> shp_x{1, 11, 20, 1, 2}; in TEST_F() local
43 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dconvert_tuple_output_to_maketuple_test.cc42 std::vector<int64_t> shp_x{5, 2, 10}; in TEST_F() local
46 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dconst_to_attr_strided_slice_grad_test.cc55 std::vector<int64_t> shp_x{16, 1, 1024}; in TEST_F() local
56 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/enhancer/
Dinsert_tensor_move_for_hccl_op_test.cc60 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
61 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
81 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
82 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
107 std::vector<int64_t> shp_x{3, 2}; in TEST_F() local
108 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
129 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
130 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
156 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
157 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fission/
Dbatch_norm_grad_infer_fission_test.cc36 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
37 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
58 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
59 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
78 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
79 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dunsorted_segment_sum_fission_test.cc35 std::vector<int64_t> shp_x{3, 39, 1}; in TEST_F() local
37 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
55 std::vector<int64_t> shp_x{3, 39, 2}; in TEST_F() local
57 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dsingle_batch_norm_fission_test.cc35 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
36 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
58 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
59 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dbatch_norm_bert_fission_test.cc34 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
74 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
75 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dbn_split_test.cc55 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
57 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
104 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
106 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dbn_grad_split_test.cc46 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
48 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
112 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
114 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dlayer_norm_grad_split_test.cc46 std::vector<int64_t> shp_x{1, 64, 112, 112}; in TEST_F() local
48 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/
Dfused_batch_norm_fusion_test.cc34 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
35 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
57 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
58 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
80 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
81 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dbatchnorm_to_bninfer_test.cc35 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
36 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
55 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
56 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
Dbatchnormgrad_to_bninfergrad_test.cc36 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
37 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
56 std::vector<int64_t> shp_x{32, 64, 112, 112}; in TEST_F() local
57 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
/third_party/mindspore/tests/ut/cpp/pre_activate/ascend/mindir/
Dall_to_all_unify_mindir_test.cc41 std::vector<int64_t> shp_x{2, 2}; in TEST_F() local
43 AbstractBasePtrList{std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x)}); in TEST_F()
60 std::vector<int64_t> shp_x{4, 2, 224, 224}; in TEST_F() local
61 auto x_abstract = std::make_shared<abstract::AbstractTensor>(kFloat32, shp_x); in TEST_F()
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc65 ShapeVector shp_x; in InferImplBroadCastShape() local
66 (void)std::transform(std::begin(shp_tuple_x), std::end(shp_tuple_x), std::back_inserter(shp_x), in InferImplBroadCastShape()
77 ShapeVector res = BroadcastShape(shp_x, shp_y); in InferImplBroadCastShape()