Home
last modified time | relevance | path

Searched refs:outer_index (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dcumsum.h48 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 Dlog_softmax.h179 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 Dlog_softmax.h45 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 Dl2normalization.h34 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 DSoftMax.cu264 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 Dttcolr.c581 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 Donedevice_train_palm_detection_filter_inference_beta.cc510 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 Donedevice_train_palm_detection_filter_inference.cc511 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 Donedevice_train_palm_detection_filter_inference_v2.cc510 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()