Home
last modified time | relevance | path

Searched refs:gemm_context (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dgemm_support.cc26 std::unique_ptr<gemmlowp::GemmContext> gemm_context; member
38 ptr->gemm_context->set_max_num_threads(context->recommended_num_threads); in Refresh()
51 ptr->gemm_context.reset(new gemmlowp::GemmContext()); in IncrementUsageCounter()
53 ptr->gemm_context->set_max_num_threads(context->recommended_num_threads); in IncrementUsageCounter()
80 return ptr->gemm_context.get(); in GetFromContext()
Dfully_connected.cc295 gemmlowp::GemmContext* gemm_context) { in FullyConnectedInt8() argument
308 GetTensorShape(output), GetTensorData<int8_t>(output), gemm_context); in FullyConnectedInt8()
318 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in EvalQuantized() local
338 gemm_context); \ in EvalQuantized()
352 FullyConnectedInt8(data, input, filter, bias, output, gemm_context); in EvalQuantized()
369 FullyConnectedInt8(data, input, filter, bias, output, gemm_context); in EvalQuantized()
394 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in EvalShuffledQuantized() local
417 GetTensorData<uint8_t>(shuffled_input_workspace), gemm_context); \ in EvalShuffledQuantized()
Dconv.cc425 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in EvalQuantized() local
465 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
492 GetTensorShape(im2col), GetTensorData<uint8_t>(im2col), gemm_context); in EvalQuantized()
Dreduce.cc298 gemmlowp::GemmContext* gemm_context = in EvalMean() local
306 op_context.output->params.scale, gemm_context); in EvalMean()
Dlstm.cc777 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in Eval() local
816 GetTensorData<int16_t>(activation_temp), gemm_context); in Eval()
/external/gemmlowp/test/
Dbenchmark_all_sizes.cc136 GemmContext gemm_context; in benchmark_8bit() local
137 gemm_context.set_max_num_threads(kNumThreads); in benchmark_8bit()
139 &gemm_context, lhs.const_map(), rhs.const_map(), &result.map(), -128, in benchmark_8bit()
150 &gemm_context, lhs.const_map(), rhs.const_map(), &result.map(), -128, in benchmark_8bit()
178 GemmContext gemm_context; in benchmark_8bit_to_32bit() local
179 gemm_context.set_max_num_threads(kNumThreads); in benchmark_8bit_to_32bit()
181 &gemm_context, lhs.const_map(), rhs.const_map(), &result.map(), -128, in benchmark_8bit_to_32bit()
192 &gemm_context, lhs.const_map(), rhs.const_map(), &result.map(), -128, in benchmark_8bit_to_32bit()
Dtest.cc779 GemmContext gemm_context; in TestWithSmallDataPerChannelQuantization() local
784 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestWithSmallDataPerChannelQuantization()
976 GemmContext gemm_context; in TestWithLargeDataPerChannelQuantization() local
981 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestWithLargeDataPerChannelQuantization()
1089 GemmContext gemm_context; in TestMultithreadedPerChannelQuantization() local
1094 &gemm_context, lhs, rhs, &result, lhs_offset, rhs_offset, in TestMultithreadedPerChannelQuantization()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dfully_connected.h70 uint8* output_data, void* gemm_context) { in FullyConnected() argument
71 (void)gemm_context; // only used in optimized code. in FullyConnected()
119 int16* output_data, void* gemm_context) { in FullyConnected() argument
120 (void)gemm_context; // only used in optimized code. in FullyConnected()
174 void* gemm_context) { in ShuffledFullyConnected() argument
175 (void)gemm_context; // only used in optimized code. in ShuffledFullyConnected()
Dlegacy_reference_ops.h284 gemmlowp::GemmContext* gemm_context) { in Conv() argument
305 output_data, DimsToShape(im2col_dims), im2col_data, gemm_context); in Conv()
318 gemmlowp::GemmContext* gemm_context) { in Conv() argument
323 im2col_data, im2col_dims, gemm_context); in Conv()
338 gemmlowp::GemmContext* gemm_context) { in Conv() argument
352 im2col_data, im2col_dims, gemm_context); in Conv()
365 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv() argument
370 output_dims, im2col_data, im2col_dims, gemm_context); in Conv()
430 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
444 gemm_context); in FullyConnected()
[all …]
Dreference_ops.h211 uint8* im2col_data, gemmlowp::GemmContext* gemm_context) { in Conv() argument
214 (void)gemm_context; // only used in optimized code. in Conv()
2053 int16* activ_temp_data_int16, gemmlowp::GemmContext* gemm_context) { in LstmCell() argument
2054 (void)gemm_context; // only used in optimized code. in LstmCell()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dfully_connected.h28 int8_t* output_data, void* gemm_context) { in FullyConnected() argument
29 (void)gemm_context; // only used in optimized code. in FullyConnected()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h364 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
378 gemm_context); in FullyConnected()
387 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
401 gemm_context); in FullyConnected()
414 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
423 output_activation_max, output_data, output_dims, gemm_context); in FullyConnected()
432 uint8* shuffled_input_workspace_data, gemmlowp::GemmContext* gemm_context) { in ShuffledFullyConnected() argument
444 shuffled_input_workspace_data, gemm_context); in ShuffledFullyConnected()
614 gemmlowp::GemmContext* gemm_context) { in Conv() argument
635 output_data, DimsToShape(im2col_dims), im2col_data, gemm_context); in Conv()
[all …]
Doptimized_ops.h1099 uint8* output_data, gemmlowp::GemmContext* gemm_context) { in FullyConnectedAsGEMV() argument
1106 gemm_context->max_num_threads(), output_rows, batches, input_size); in FullyConnectedAsGEMV()
1134 gemm_context->workers_pool()->Execute(tasks); in FullyConnectedAsGEMV()
1171 uint8* output_data, gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1199 output_activation_max, output_shape, output_data, gemm_context); in FullyConnected()
1221 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1230 int16* output_data, gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1241 (void)gemm_context; // only used in properly optimized code. in FullyConnected()
1308 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1637 gemmlowp::GemmContext* gemm_context) { in ShuffledFullyConnected() argument
[all …]
/external/gemmlowp/doc/
Dquantization_example.cc360 gemmlowp::GemmContext gemm_context; in main() local
363 &gemm_context, uint8_lhs.ConstMap(), uint8_rhs.ConstMap(), in main()
Dpublic.md48 &gemm_context, uint8_lhs_matrix, uint8_rhs_matrix,
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h1154 uint8* output_data, gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1202 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1216 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1230 gemm_context); in FullyConnected()
1238 int16* output_data, gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1249 (void)gemm_context; // only used in properly optimized code. in FullyConnected()
1316 gemm_context, weights_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
1328 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
1342 gemm_context); in FullyConnected()
1356 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
[all …]