Searched defs:GetOutputDeviceShape (Results 1 – 2 of 2) sorted by relevance
357 TEST_F(AnfRuntimeAlgorithmTest, GetOutputDeviceShape) { in TEST_F() argument
876 std::vector<size_t> AnfRuntimeAlgorithm::GetOutputDeviceShape(const AnfNodePtr &node, size_t output… in GetOutputDeviceShape() function in mindspore::session::AnfRuntimeAlgorithm