Searched defs:GetOutputShape (Results 1 – 3 of 3) sorted by relevance
67 std::vector<size_t> CacheManager::GetOutputShape(const AnfNodePtr &node, size_t index) { in GetOutputShape() function in mindspore::opt::CacheManager
58 std::vector<int64_t> GetOutputShape(const std::vector<int64_t> &in_shape, int64_t kernel_d, int64_t… in GetOutputShape() function
162 std::vector<size_t> AkgKernelJsonGenerator::GetOutputShape(const AnfNodePtr &anf_node, size_t index… in GetOutputShape() function in mindspore::kernel::AkgKernelJsonGenerator