Home
last modified time | relevance | path

Searched refs:src_data_ptr (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/
H A Dsparsity_format_converter.cc300 int level, int prev_idx, int* src_data_ptr, in Populate() argument
320 src_data[*src_data_ptr]; in Populate()
322 *src_data_ptr = *src_data_ptr + 1; in Populate()
332 src_data_ptr, dest_data); in Populate()
341 Populate(src_data, indices, level + 1, i, src_data_ptr, dest_data); in Populate()
353 int src_data_ptr = 0; in SparseToDense() local
355 Populate(src_data, indices, 0, 0, &src_data_ptr, data_.data()); in SparseToDense()
378 int src_data_ptr = 0; in SparseToDense() local
380 Populate(src_data, indices, 0, 0, &src_data_ptr, dest_data); in SparseToDense()
H A Dsparsity_format_converter.h111 int prev_idx, int* src_data_ptr, T* dest_data);
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dshuffle_fc_weights.cc126 const uint8* src_data_ptr = weights_data.data() + (r + i) * cols + c; in Run() local
128 uint8 src_val = *src_data_ptr++; in Run()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dfully_connected.h205 const uint8_t* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
207 uint8_t src_val = *src_data_ptr++; in ShuffledFullyConnected()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h804 const uint8* src_data_ptr = input_data + c; in ShuffledFullyConnected() local
805 uint8x16_t val0 = vld1q_u8(src_data_ptr + 0 * accum_depth); in ShuffledFullyConnected()
806 uint8x16_t val1 = vld1q_u8(src_data_ptr + 1 * accum_depth); in ShuffledFullyConnected()
807 uint8x16_t val2 = vld1q_u8(src_data_ptr + 2 * accum_depth); in ShuffledFullyConnected()
808 uint8x16_t val3 = vld1q_u8(src_data_ptr + 3 * accum_depth); in ShuffledFullyConnected()
822 const uint8* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
824 uint8 src_val = *src_data_ptr++; in ShuffledFullyConnected()
6412 const float* src_data_ptr = input_data + i; in AffineQuantize() local
6413 float32x4_t input_val_0 = vld1q_f32(src_data_ptr); in AffineQuantize()
6414 float32x4_t input_val_1 = vld1q_f32(src_data_ptr + 4); in AffineQuantize()
[all …]
H A Dlegacy_optimized_ops.h2272 const uint8* src_data_ptr = input_data + c; in ShuffledFullyConnected() local
2273 uint8x16_t val0 = vld1q_u8(src_data_ptr + 0 * accum_depth); in ShuffledFullyConnected()
2274 uint8x16_t val1 = vld1q_u8(src_data_ptr + 1 * accum_depth); in ShuffledFullyConnected()
2275 uint8x16_t val2 = vld1q_u8(src_data_ptr + 2 * accum_depth); in ShuffledFullyConnected()
2276 uint8x16_t val3 = vld1q_u8(src_data_ptr + 3 * accum_depth); in ShuffledFullyConnected()
2290 const uint8* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
2292 uint8 src_val = *src_data_ptr++; in ShuffledFullyConnected()