Home
last modified time | relevance | path

Searched refs:result_shp (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc539 ShapeVector result_shp; in InferImplPad() local
545 result_shp.push_back(input_shp[i] + paddings_vec[i][0] + paddings_vec[i][1]); in InferImplPad()
547 return std::make_shared<AbstractTensor>(arg->element(), std::make_shared<Shape>(result_shp)); in InferImplPad()
Dprim_arrays.cc735 ShapeVector result_shp; in InferImplTranspose() local
743 result_shp.push_back(input_shp[idx]); in InferImplTranspose()
747 …return std::make_shared<AbstractTensor>(input->element(), std::make_shared<Shape>(result_shp, min_… in InferImplTranspose()