Searched refs:shape_int (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/ |
| D | trt_converter_context.cc | 274 ShapeVector shape_int; in CreateTempWeight() local
275 std::transform(shape.begin(), shape.end(), std::back_inserter(shape_int), SizeToLong); in CreateTempWeight()
276 auto tensor = std::make_shared<tensor::Tensor>(type, shape_int); in CreateTempWeight()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 1257 ShapeVector shape_int; in SetOutputInferTypeAndShape() local
1262 std::transform(shapes[0].begin(), shapes[0].end(), std::back_inserter(shape_int), SizeToLong); in SetOutputInferTypeAndShape()
1264 … std::make_shared<abstract::Shape>(shape_int, min_shape0, max_shape0)); in SetOutputInferTypeAndShape()
1266 abstract = std::make_shared<AbstractTensor>(TypeIdToType(types[0]), shape_int); in SetOutputInferTypeAndShape()
1273 ShapeVector shape_int; in SetOutputInferTypeAndShape() local
1278 … std::transform(shapes[i].begin(), shapes[i].end(), std::back_inserter(shape_int), SizeToLong); in SetOutputInferTypeAndShape()
1280 … std::make_shared<abstract::Shape>(shape_int, min_shape, max_shape)); in SetOutputInferTypeAndShape()
1283 …:make_shared<AbstractTensor>(TypeIdToType(types[i]), std::make_shared<abstract::Shape>(shape_int)); in SetOutputInferTypeAndShape()
|