Home
last modified time | relevance | path

Searched defs:shape_in (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dreshape_tests.cc32 std::vector<int> shape_in = {1, 1, 1, 7}; variable
44 std::vector<int> shape_in = {1, 2, 2, 3}; variable
56 std::vector<int> shape_in = {1, 2, 2, 4}; variable
68 std::vector<int> shape_in = {1, 4, 2, 3}; variable
80 std::vector<int> shape_in = {1, 2, 2, 5}; variable
92 std::vector<int> shape_in = {1, 4, 2, 5}; variable
110 std::vector<int> shape_in = {1, 5, 5, 8}; variable
141 std::vector<int> shape_in = {1, 3, 2, 5}; variable
155 std::vector<int> shape_in = {5, 3, 8}; variable
/third_party/mindspore/tests/st/ops/custom_ops_tbe/
Dconv_layer.py34 def conv_layer_cce_para_check(shape_in, shape_w, in_dtype, w_dtype, res_dtype, padh, padw, argument
150 def conv_layer_cce(shape_in, shape_w, in_dtype, w_dtype, res_dtype, padh, padw, strideh, stridew, argument
/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/ccsrc/minddata/dataset/engine/cache/
Dcache_fbb.cc107 auto shape_in = col_ts->dims(); in RestoreOneTensor() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc480 int64_t SlicePreposePass::GetReshapeAbnormalAxeIn(const std::vector<int64_t> &shape_in, in GetReshapeAbnormalAxeIn()
556 … const CNodePtr &reshape_cnode, const std::vector<int64_t> &shape_in, in PreposeWithNormalReshape()
622 const std::vector<int64_t> &shape_in, in CreateSlice1ForReshapePrepose()
705 … const CNodePtr &matmul_cnode, const std::vector<int64_t> &shape_in, in PreposeWithAbnormalReshape()
916 auto shape_in = GetCNodeInputShape(reshape_cnode, 1); in PreposeWithReshape() local
1083 auto shape_in = GetCNodeInputShape(fc_cnode, 1); in PreposeWithFullConnection() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dreshape_info.cc211 Arrangement shape_in; in InferTensorLayout() local