Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h44 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnected() local
48 for (int d = 0; d < accum_depth; ++d) { in FullyConnected()
49 total += input_data[b * accum_depth + d] * in FullyConnected()
50 weights_data[out_c * accum_depth + d]; in FullyConnected()
89 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
93 for (int d = 0; d < accum_depth; ++d) { in FullyConnected()
94 int32_t input_val = input_data[b * accum_depth + d]; in FullyConnected()
95 int32_t filter_val = filter_data[out_c * accum_depth + d]; in FullyConnected()
136 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
143 for (int d = 0; d < accum_depth; ++d) { in FullyConnected()
[all …]
Dbatch_matmul.h78 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
95 for (int k = 0; k < accum_depth; ++k) { in BatchMatMul()
97 lhs_ptr2[accum_depth * i + k] * rhs_ptr2[j * accum_depth + k]; in BatchMatMul()
136 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
151 lhs_data, row_sums, num_weights_matrices * lhs_rows, accum_depth); in BatchMatMul()
183 for (int k = 0; k < accum_depth; ++k) { in BatchMatMul()
185 lhs_ptr2[accum_depth * i + k] * rhs_ptr2[j * accum_depth + k]; in BatchMatMul()
225 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
252 for (int k = 0; k < accum_depth; ++k) { in BatchMatMul()
253 AccumT lhs_val = lhs_ptr2[accum_depth * i + k]; in BatchMatMul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h44 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
48 for (int d = 0; d < accum_depth; ++d) { in FullyConnected()
49 int32_t input_val = input_data[b * accum_depth + d]; in FullyConnected()
50 int32_t filter_val = filter_data[out_c * accum_depth + d]; in FullyConnected()
84 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
88 for (int d = 0; d < accum_depth; ++d) { in FullyConnected()
89 int32_t input_val = input_data[b * accum_depth + d]; in FullyConnected()
90 int32_t filter_val = filter_data[out_c * accum_depth + d]; in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h78 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
83 lhs_params.cols = accum_depth; in BatchMatMul()
87 rhs_params.rows = accum_depth; in BatchMatMul()
169 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
184 lhs_data, row_sums, num_weights_matrices * lhs_rows, accum_depth); in BatchMatMul()
193 lhs_params.cols = accum_depth; in BatchMatMul()
197 rhs_params.rows = accum_depth; in BatchMatMul()
323 const int accum_depth = extended_lhs_shape.Dims(4); in BatchMatMul() local
337 lhs_params.cols = accum_depth; in BatchMatMul()
342 rhs_params.rows = accum_depth; in BatchMatMul()
Dlegacy_optimized_ops.h1666 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
1676 if (filter_offset == -128 && !(output_depth % 4) && !(accum_depth % 64)) { in FullyConnected()
1683 if (!(output_depth % 4) && !(accum_depth % 8)) { in FullyConnected()
1693 filter_data, output_depth, accum_depth); in FullyConnected()
1695 input_data, accum_depth, batches); in FullyConnected()
2179 int output_stride, int accum_depth, in LegacyShuffledFullyConnectedWorkerTask()
2188 accum_depth_(accum_depth), in LegacyShuffledFullyConnectedWorkerTask()
2241 const int accum_depth = weights_shape.Dims(weights_dim_count - 1); in ShuffledFullyConnected() local
2242 TFLITE_DCHECK((accum_depth % 16) == 0); in ShuffledFullyConnected()
2255 for (int i = 0; i < accum_depth; i += 16) { in ShuffledFullyConnected()
[all …]
Doptimized_ops.h460 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
464 lhs_params.cols = accum_depth; in FullyConnected()
470 rhs_params.rows = accum_depth; in FullyConnected()
500 int output_stride, int accum_depth, const int32* bias_data, in ShuffledFullyConnectedWorkerImpl() argument
513 for (int d = 0; d < accum_depth; d += 16) { in ShuffledFullyConnectedWorkerImpl()
594 for (int d = 0; d < accum_depth; d += 16) { in ShuffledFullyConnectedWorkerImpl()
688 for (int d = 0; d < accum_depth; d += 16) { in ShuffledFullyConnectedWorkerImpl()
734 for (int d = 0; d < accum_depth; d += 16) { in ShuffledFullyConnectedWorkerImpl()
777 int output_stride, int accum_depth, in ShuffledFullyConnectedWorkerTask()
786 accum_depth_(accum_depth), in ShuffledFullyConnectedWorkerTask()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dfully_connected.cc51 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in FullyConnected() local
52 const int accum_depth_iters = accum_depth / 2; in FullyConnected()
64 const int8_t* input_ptr = (input_data + b * accum_depth) - 2; in FullyConnected()
65 const int8_t* filter_ptr = (filter_data + out_c * accum_depth) - 2; in FullyConnected()
212 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in EvalQuantizedInt8()
221 (tflite::micro::GetTensorData<int8_t>(input) + b * accum_depth), in EvalQuantizedInt8()
222 tflite::micro::GetTensorData<int32_t>(bias), accum_depth, in EvalQuantizedInt8()
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dfully_connected.cc114 const int accum_depth = filter_shape.Dims(filter_dim_count - 1); in EvalQuantizedInt8() local
132 input_dims.c = accum_depth; in EvalQuantizedInt8()
135 filter_dims.n = accum_depth; in EvalQuantizedInt8()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h47 const int accum_depth = weights_shape.Dims(weights_dims_count - 1); in FullyConnectedSparseWeight() local
61 weights_data[pw1] * input_data[b * accum_depth + idx_1]; in FullyConnectedSparseWeight()