Searched defs:outer_index (Results 1 – 9 of 9) sorted by relevance
48 for (size_t outer_index = 0; outer_index < outer; outer_index++) { in CumSum() local116 for (size_t outer_index = 0; outer_index < outer; outer_index++) { in CumSum() local
179 for (size_t outer_index = 0; outer_index < outer_size; ++outer_index) { in LogSoftmaxQuantized() local
34 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in L2Normalization() local
45 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in LogSoftmax() local
264 for (index_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in cunn_SpatialSoftMaxForward() local322 for (uint32_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in cunn_SpatialSoftMaxBackward() local
510 const Tidx* __restrict outer_index, in SparseDenseMatMulCSR()529 const Tidx* __restrict outer_index, in SparseFullyConnectedCSR()
511 const Tidx* __restrict outer_index, in SparseDenseMatMulCSR()530 const Tidx* __restrict outer_index, in SparseFullyConnectedCSR()
581 FT_UInt outer_index = 0; in get_deltas_for_var_index_base() local