Searched defs:ostartH (Results 1 – 5 of 5) sorted by relevance
69 int ostartH = blockDim.y*blockIdx.y + threadIdx.y; in adaptive_average_pool() local139 int ostartH = START_IND(ih, isizeH, osizeH); in adaptive_average_gradinput() local184 int ostartH = blockDim.y*blockIdx.y + threadIdx.y; in atomic_adaptive_average_gradinput() local272 index_t ostartH = threadIdx.z + blockIdx.z*oH; in C10_LAUNCH_BOUNDS_1() local405 index_t ostartH = START_IND_INT(ih, isizeH, osizeH); in C10_LAUNCH_BOUNDS_1() local
62 int ostartH = blockDim.y*blockIdx.y + threadIdx.y; in adaptivemaxpool() local128 int ostartH = blockDim.y*blockIdx.y + threadIdx.y; in adaptivemaxgradinput() local175 int ostartH = blockDim.y*blockIdx.y + threadIdx.y; in atomicadaptivemaxgradinput() local
58 int ostartH = blockIdx.y * blockDim.y + threadIdx.y; in adaptivemaxpool() local171 int ostartH = blockIdx.y * blockDim.y + threadIdx.y; in adaptivemaxgradinput() local247 int ostartH = blockIdx.y * blockDim.y + threadIdx.y; in atomicadaptivemaxgradinput() local
63 int ostartH = blockIdx.y * blockDim.y + threadIdx.y; in adaptiveaveragepool() local186 int ostartH = start_index(ih, isizeH, osizeH); in adaptiveaveragegradinput() local261 int ostartH = blockIdx.y * blockDim.y + threadIdx.y; in atomicadaptiveaveragegradinput() local
124 int ostartH = threadIdx.z + blockIdx.z*oH; in C10_LAUNCH_BOUNDS_1() local