/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()
|