Searched refs:int_shape (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | gpu_session.cc | 616 ShapeVector int_shape; in UpdateOutputTensors() local
617 …std::transform(updated_shape.begin(), updated_shape.end(), std::back_inserter(int_shape), SizeToIn… in UpdateOutputTensors()
618 tensor->set_shape(int_shape); in UpdateOutputTensors()
|
| D | session_basic.cc | 1639 ShapeVector int_shape; in UpdateOutputs() local
1640 …(void)std::transform(updated_shape.begin(), updated_shape.end(), std::back_inserter(int_shape), Si… in UpdateOutputs()
1641 (void)tensor->set_shape(int_shape); in UpdateOutputs()
1741 ShapeVector int_shape; in UpdateOutputTensors() local
1742 …d::transform(updated_shape.begin(), updated_shape.end(), std::back_inserter(int_shape), SizeToInt); in UpdateOutputTensors()
1743 (void)tensor->set_shape(int_shape); in UpdateOutputTensors()
|
| D | ascend_session.cc | 2051 ShapeVector int_shape; in UpdateOutputTensors() local
2052 …d::transform(updated_shape.begin(), updated_shape.end(), std::back_inserter(int_shape), SizeToInt); in UpdateOutputTensors()
2053 (void)tensor->set_shape(int_shape); in UpdateOutputTensors()
|