Home
last modified time | relevance | path

Searched defs:shape_out (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dreshape_tests.cc33 std::vector<int> shape_out = {1, 7}; variable
45 std::vector<int> shape_out = {1, 1, 4, 3}; variable
57 std::vector<int> shape_out = {4, 4}; variable
69 std::vector<int> shape_out = {1, 3, 2, 4}; variable
81 std::vector<int> shape_out = {1, 1, 5, 4}; variable
93 std::vector<int> shape_out = {1, 2, 5, 4}; variable
111 std::vector<int> shape_out = {8, 1, 5, 5}; variable
142 std::vector<int> shape_out = {1, 5, 2, 3}; variable
156 std::vector<int> shape_out = {8, 15}; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Ddynamic_creator.h51 OperatorInfoPtr Create(const std::string &name, const Shapes &shape_in, const Shapes &shape_out, in Create()
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc481 const std::vector<int64_t> &shape_out, in GetReshapeAbnormalAxeIn()
508 const std::vector<int64_t> &shape_out, in GetReshapeAbnormalIndexOut()
706const std::vector<int64_t> &shape_out, const int64_t abnormal_axe_in, in PreposeWithAbnormalReshape()
917 auto shape_out = GetCNodeInputShape(slice_cnode, 1); in PreposeWithReshape() local
1084 auto shape_out = GetCNodeInputShape(slice_cnode, 1); in PreposeWithFullConnection() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreshape_info.cc228 Arrangement shape_out; in InferTensorLayout() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc101 ShapeVector shape_out = {input_shape->shape()[0], input_shape->shape()[1], h_out, w_out}; in InferImplPooling() local