Searched defs:outer_index (Results 1 – 11 of 11) sorted by relevance
122 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumprodCompute() local156 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumprodCompute() local
84 int64_t outer_index = i * dim_on * num_inner; in ExecArgMinWithValue() local
147 for (int64_t outer_index = 0; outer_index < outer; ++outer_index) { in ReduceSumOneAxes() local224 for (int64_t outer_index = 0; outer_index < outer; outer_index++) { in ReduceSumOneAxes2() local
79 int64_t outer_index = i * dim_on * num_inner; in ExecArgMaxWithValue() local
95 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumulativeProcess() local
79 for (size_t outer_index = 0; outer_index < outer; ++outer_index) { in CumulativeProcess() local
334 size_t outer_index = index / inner_size_; in SetOffset() local339 inline void SetOffset(size_t outer_index, size_t inner_index) { in SetOffset()
385 outer_index: ty::DebruijnIndex, in has_late_bound_regions() field
124 outer_index: ty::DebruijnIndex, field
197 outer_index: ty::DebruijnIndex, in any_free_region_meets() field412 outer_index: ty::DebruijnIndex, field
140 for (uint32_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in SpatialSoftMaxForward() local