Searched refs:tensor_out_shape (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/core/ir/ |
| D | pattern_matcher.h | 905 ShapeVector tensor_out_shape = new_tensor_ptr->shape();
906 …int data_out_size = std::accumulate(tensor_out_shape.begin(), tensor_out_shape.end(), 1, std::mult…
972 auto tensor_out_shape = tensor_1_abstract->shape()->shape();
974 return std::make_shared<tensor::Tensor>(tensor_1_type_ptr->type_id(), tensor_out_shape);
984 auto tensor_out_shape = tensor_3_abstract->shape()->shape();
985 …int data_out_size = std::accumulate(tensor_out_shape.begin(), tensor_out_shape.end(), 1, std::mult…
992 return std::make_shared<tensor::Tensor>(tensor_3_type_ptr->type_id(), tensor_out_shape);
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_others.cc | 424 ShapeVector tensor_out_shape = {Shape::SHP_ANY, tensor_in_shape[1]}; in InferImplAllSwap() local
430 … std::make_shared<Shape>(tensor_out_shape, min_shape, max_shape)); in InferImplAllSwap()
433 tensor_out->element(), std::make_shared<Shape>(tensor_out_shape, min_shape, max_shape)); in InferImplAllSwap()
|