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 D | sparsity_format_converter.cc | 300 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 D | sparsity_format_converter.h | 111 int prev_idx, int* src_data_ptr, T* dest_data);
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | shuffle_fc_weights.cc | 126 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 D | fully_connected.h | 205 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 D | optimized_ops.h | 804 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 D | legacy_optimized_ops.h | 2272 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()
|