Searched defs:GetOutputDeviceShape (Results 1 – 2 of 2) sorted by relevance
642 ShapeVector AnfRuntimeAlgorithm::GetOutputDeviceShape(const AnfNodePtr &node, size_t output_idx) { in GetOutputDeviceShape() function in mindspore::session::AnfRuntimeAlgorithm657 ShapeVector AnfRuntimeAlgorithm::GetOutputDeviceShape(const AnfNodePtr &node, size_t output_idx, in GetOutputDeviceShape() function in mindspore::session::AnfRuntimeAlgorithm
373 TEST_F(AnfRuntimeAlgorithmTest, GetOutputDeviceShape) { in TEST_F() argument