Searched defs:outer_idx (Results 1 – 11 of 11) sorted by relevance
287 int64_t outer_idx = i / num_chunks; in _vec_softmax_backward() local379 int64_t outer_idx = i / num_chunks; in _vec_softmax_backward() local498 int64_t outer_idx = i / num_chunks; in _vec_log_softmax_backward() local586 int64_t outer_idx = i / num_chunks; in _vec_log_softmax_backward() local711 int64_t outer_idx = idx / inner_size; in _vec_softmax() local814 int64_t outer_idx = idx / inner_size; in _vec_softmax() local914 int64_t outer_idx = i / num_chunks; in _vec_logsoftmax() local1024 int64_t outer_idx = i / num_chunks; in _vec_logsoftmax() local
65 SliceIndex outer_idx = static_cast<SliceIndex>(r_start / indices_size); in HandleCopiesBatched() local
55 for (size_t outer_idx = 0; outer_idx < outer_size; ++outer_idx) { in log_softmax_kernel() local
56 for (const auto outer_idx : c10::irange(outer_offset)) { in _segment_reduce_lengths_cpu_kernel1() local228 for (const auto outer_idx : c10::irange(outer_offset)) { in _segment_reduce_cpu_lengths_backward_kernel1() local
186 int64_t outer_idx = i / inner_size; in __anon010efb6d0202() local293 int64_t outer_idx = i / inner_size; in host_softmax_backward() local
110 int64_t outer_idx = row_id / segment_count; in segment_reduce_forward_kernel() local181 int64_t outer_idx = row_id / segment_count; in segment_reduce_backward_kernel() local
88 for (size_t outer_idx = 0; outer_idx < outer_size; ++outer_idx) { in perform_topk() local
225 for (size_t outer_idx = 0; outer_idx < outer_size; ++outer_idx) { in apply_over_dim() local
586 let outer_idx = graph.to_index(outer); in augment_path() localVariable
887 final int outer_idx = ic.getOuterClassIndex(); in visitInnerClasses() local
383 size_t outer_idx = index_pair.index() / 2; in FoldOperandsPermutation() local