Searched defs:inner_index (Results 1 – 9 of 9) sorted by relevance
47 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local54 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local82 for (int inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmax() local
39 for (int inner_index = 0; inner_index < depth; ++inner_index) { in L2Normalization() local49 for (int inner_index = 0; inner_index < depth; ++inner_index) { in L2Normalization() local
181 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local188 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local217 for (size_t inner_index = 0; inner_index < depth; ++inner_index) { in LogSoftmaxQuantized() local
54 for (size_t inner_index = 0; inner_index < inner; inner_index++) { in CumSum() local122 for (size_t inner_index = 0; inner_index < inner; inner_index++) { in CumSum() local
157 Value inner_index = rewriter.create<LLVM::ConstantOp>( in convertFuncOpToLLVMFuncOp() local
266 …for (index_t inner_index = blockIdx.y * blockDim.y + threadIdx.y; inner_index < inner_size; inner_… in cunn_SpatialSoftMaxForward() local324 …for (uint32_t inner_index = blockIdx.y * blockDim.y + threadIdx.y; inner_index < inner_size; inner… in cunn_SpatialSoftMaxBackward() local
61 int64_t inner_index = GetIndexForId(index->operand(1), id, map_id); in GetIndexForId() local
5016 HloInstruction* inner_index = operand->mutable_operand(i); in HandleDynamicSlice() local5138 HloInstruction* inner_index = dus_update->mutable_operand(i); in HandleDynamicUpdateSlice() local
582 FT_UInt inner_index = 0; in get_deltas_for_var_index_base() local