Searched defs:first_dim (Results 1 – 3 of 3) sorted by relevance
75 int64_t Util::LocalShard(int64_t first_dim, int64_t rank_id, int64_t server_num) { in LocalShard()83 std::map<int64_t, int64_t> Util::AllRankLocalShard(int64_t first_dim, int64_t rank_id, int64_t serv… in AllRankLocalShard()
174 …const std::pair<int64_t, int64_t> first_dim = {(c0.first * range[kH].first * range[kW].first + k16… in FracZC04Range() local190 …const std::pair<int64_t, int64_t> first_dim = {(i.first + k16 - 1) / k16 + (h.first + k16 - 1) / k… in FracZNLSTMCRange() local
471 auto first_dim = DivCeil(c0 * shape[kH] * shape[kW], kCubeSize); in FracZc04DeviceShape() local487 int64_t first_dim; in FracZc04DeviceDynamicShape() local