Searched defs:MakeInputShape (Results 1 – 1 of 1) sorted by relevance
186 std::array<int64_t, 2> MakeInputShape(int64_t /*D*/, int64_t H, int64_t W) { in MakeInputShape() function190 std::array<int64_t, 3> MakeInputShape(int64_t D, int64_t H, int64_t W) { in MakeInputShape() function