| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | cumsum.h | 48 for (size_t outer_index = 0; outer_index < outer; outer_index++) { in CumSum() local
51 outer_index_adj = (outer - 1) - outer_index; in CumSum()
53 outer_index_adj = outer_index; in CumSum()
116 for (size_t outer_index = 0; outer_index < outer; outer_index++) { in CumSum() local
119 outer_index_adj = (outer - 1) - outer_index; in CumSum()
121 outer_index_adj = outer_index; in CumSum()
|
| H A D | log_softmax.h | 179 for (size_t outer_index = 0; outer_index < outer_size; ++outer_index) { in LogSoftmaxQuantized() local
183 std::max(max_in_row, input_data[outer_index * depth + inner_index]); in LogSoftmaxQuantized()
190 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmaxQuantized()
219 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmaxQuantized()
236 output_data[outer_index * depth + inner_index] = in LogSoftmaxQuantized()
239 output_data[outer_index * depth + inner_index] = kMinT8; in LogSoftmaxQuantized()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| H A D | log_softmax.h | 45 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in LogSoftmax() local
49 std::max(max_in_row, input_data[outer_index * depth + inner_index]); in LogSoftmax()
56 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmax()
84 static_cast<int32_t>(input_data[outer_index * depth + inner_index]) - in LogSoftmax()
101 output_data[outer_index * depth + inner_index] = in LogSoftmax()
104 output_data[outer_index * depth + inner_index] = kMinInt8; in LogSoftmax()
|
| H A D | l2normalization.h | 34 for (int outer_index = 0; outer_index < outer_size; ++outer_index) { in L2Normalization() local
41 input_data[depth * outer_index + inner_index] - input_zero_point; in L2Normalization()
51 input_data[depth * outer_index + inner_index] - input_zero_point; in L2Normalization()
59 output_data[depth * outer_index + inner_index] = in L2Normalization()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | SoftMax.cu | 264 for (index_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in cunn_SpatialSoftMaxForward() local
265 const index_t outer_offset = outer_index * outer_stride; in cunn_SpatialSoftMaxForward()
322 for (uint32_t outer_index = blockIdx.x; outer_index < outer_size; outer_index += gridDim.x) { in cunn_SpatialSoftMaxBackward() local
323 const uint32_t outer_offset = outer_index * outer_stride; in cunn_SpatialSoftMaxBackward()
|
| /aosp_15_r20/external/freetype/src/sfnt/ |
| H A D | ttcolr.c | 581 FT_UInt outer_index = 0; in get_deltas_for_var_index_base() local
606 outer_index = colr->delta_set_idx_map.outerIndex[loop_var_index]; in get_deltas_for_var_index_base()
611 outer_index = 0; in get_deltas_for_var_index_base()
616 outer_index, inner_index ); in get_deltas_for_var_index_base()
|
| /aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/ |
| H A D | onedevice_train_palm_detection_filter_inference_beta.cc | 510 const Tidx* __restrict outer_index, in SparseDenseMatMulCSR() argument
518 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseDenseMatMulCSR()
529 const Tidx* __restrict outer_index, in SparseFullyConnectedCSR() argument
540 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseFullyConnectedCSR()
|
| H A D | onedevice_train_palm_detection_filter_inference.cc | 511 const Tidx* __restrict outer_index, in SparseDenseMatMulCSR() argument
519 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseDenseMatMulCSR()
530 const Tidx* __restrict outer_index, in SparseFullyConnectedCSR() argument
541 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseFullyConnectedCSR()
|
| H A D | onedevice_train_palm_detection_filter_inference_v2.cc | 510 const Tidx* __restrict outer_index, in SparseDenseMatMulCSR() argument
518 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseDenseMatMulCSR()
529 const Tidx* __restrict outer_index, in SparseFullyConnectedCSR() argument
540 num_rows, num_cols, nnz_shape[0], outer_index, cols, nnz_values); in SparseFullyConnectedCSR()
|