Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter.cc260 int level, int prev_idx, int* src_data_ptr, in Populate() argument
280 src_data[*src_data_ptr]; in Populate()
282 *src_data_ptr = *src_data_ptr + 1; in Populate()
292 src_data_ptr, dest_data); in Populate()
300 Populate(src_data, indices, level + 1, i, src_data_ptr, dest_data); in Populate()
311 int src_data_ptr = 0; in SparseToDense() local
313 Populate(src_data, indices, 0, 0, &src_data_ptr, data_.data()); in SparseToDense()
336 int src_data_ptr = 0; in SparseToDense() local
338 Populate(src_data, indices, 0, 0, &src_data_ptr, dest_data); in SparseToDense()
Dformat_converter.h77 int prev_idx, int* src_data_ptr, T* dest_data);
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dshuffle_fc_weights.cc125 const uint8* src_data_ptr = weights_data.data() + (r + i) * cols + c; in Run() local
127 uint8 src_val = *src_data_ptr++; in Run()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h202 const uint8_t* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
204 uint8_t src_val = *src_data_ptr++; in ShuffledFullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h868 const uint8* src_data_ptr = input_data + c; in ShuffledFullyConnected() local
869 uint8x16_t val0 = vld1q_u8(src_data_ptr + 0 * accum_depth); in ShuffledFullyConnected()
870 uint8x16_t val1 = vld1q_u8(src_data_ptr + 1 * accum_depth); in ShuffledFullyConnected()
871 uint8x16_t val2 = vld1q_u8(src_data_ptr + 2 * accum_depth); in ShuffledFullyConnected()
872 uint8x16_t val3 = vld1q_u8(src_data_ptr + 3 * accum_depth); in ShuffledFullyConnected()
886 const uint8* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
888 uint8 src_val = *src_data_ptr++; in ShuffledFullyConnected()
6900 const float* src_data_ptr = input_data + i; in AffineQuantize() local
6901 float32x4_t input_val_0 = vld1q_f32(src_data_ptr); in AffineQuantize()
6902 float32x4_t input_val_1 = vld1q_f32(src_data_ptr + 4); in AffineQuantize()
[all …]
Dlegacy_optimized_ops.h2271 const uint8* src_data_ptr = input_data + c; in ShuffledFullyConnected() local
2272 uint8x16_t val0 = vld1q_u8(src_data_ptr + 0 * accum_depth); in ShuffledFullyConnected()
2273 uint8x16_t val1 = vld1q_u8(src_data_ptr + 1 * accum_depth); in ShuffledFullyConnected()
2274 uint8x16_t val2 = vld1q_u8(src_data_ptr + 2 * accum_depth); in ShuffledFullyConnected()
2275 uint8x16_t val3 = vld1q_u8(src_data_ptr + 3 * accum_depth); in ShuffledFullyConnected()
2289 const uint8* src_data_ptr = input_data + b * accum_depth + c; in ShuffledFullyConnected() local
2291 uint8 src_val = *src_data_ptr++; in ShuffledFullyConnected()