Searched defs:x_shp (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ops/_utils/ |
| D | utils.py | 78 def get_concat_offset(x_shp, x_type, axis, prim_name): argument
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 134 std::vector<int64_t> x_shp{2, 16, 64}; in TEST_F() local
192 std::vector<int64_t> x_shp{2, 3}; in TEST_F() local
317 std::vector<int64_t> x_shp{10, 32, 32, 32}; in TEST_F() local
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | mat_mul.cc | 30 auto x_shp = x_shape_map[kShape]; in MatMulInferShape() local
|
| D | batch_matmul.cc | 37 auto x_shp = x_shape_map[kShape]; in BatchMatmulInferShape() local
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_maths.cc | 277 auto x_shp = x->shape()->shape(); in InferImplMatMul() local
343 auto x_shp = x->shape()->shape(); in InferImplBatchMatMul() local
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/buffer_fusion/ |
| D | buffer_fusion_test.cc | 292 std::vector<int64_t> x_shp{2048, 768}; in TEST_F() local
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 397 def _regenerate_output_shape(x_shp, ind_shp, axis): argument
|
| D | grad_nn_ops.py | 546 def _get_transpose_axis(x_shp, axis): argument
|