Searched defs:neighbor_data_dimensions (Results 1 – 1 of 1) sorted by relevance
141 const int64 neighbor_data_dimensions = warp_dims + 2; in Gather2by2Neighbors() local171 const int64 neighbor_data_dimensions = warp_dims + 2; in ScatterToGradData() local