| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | gemm_support.cc | 28 auto* ptr = reinterpret_cast<RefCountedGemmContext*>(context->gemm_context); in IncrementUsageCounter()
33 context->gemm_context = ptr; in IncrementUsageCounter()
39 auto* ptr = reinterpret_cast<RefCountedGemmContext*>(context->gemm_context); in DecrementUsageCounter()
48 context->gemm_context = nullptr; in DecrementUsageCounter()
53 auto* ptr = reinterpret_cast<RefCountedGemmContext*>(context->gemm_context); in GetFromContext()
|
| D | fully_connected.cc | 182 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in EvalQuantized() local
194 GetTensorData<uint8_t>(output), GetTensorDims(output), gemm_context) in EvalQuantized()
|
| D | conv.cc | 296 gemmlowp::GemmContext* gemm_context = gemm_support::GetFromContext(context); in EvalQuantized() local
313 GetTensorDims(im2col), gemm_context); in EvalQuantized()
328 GetTensorDims(im2col), gemm_context); in EvalQuantized()
|
| /external/gemmlowp/test/ |
| D | benchmark_all_sizes.cc | 136 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()
|
| D | test.cc | 779 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/gemmlowp/doc/ |
| D | quantization_example.cc | 360 gemmlowp::GemmContext gemm_context; in main() local
363 &gemm_context, uint8_lhs.ConstMap(), uint8_rhs.ConstMap(), in main()
|
| D | public.md | 48 &gemm_context, uint8_lhs_matrix, uint8_rhs_matrix,
|
| /external/tensorflow/tensorflow/contrib/lite/ |
| D | context.h | 293 void* gemm_context; member
|
| D | interpreter.cc | 79 context_.gemm_context = nullptr; in Interpreter()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
| D | reference_ops.h | 258 gemmlowp::GemmContext* gemm_context) { in Conv() argument
261 (void)gemm_context; // only used in optimized code. in Conv()
328 gemmlowp::GemmContext* gemm_context) { in Conv() argument
342 im2col_data, im2col_dims, gemm_context); in Conv()
355 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv() argument
360 output_dims, im2col_data, im2col_dims, gemm_context); in Conv()
501 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
502 (void)gemm_context; // only used in optimized code. in FullyConnected()
546 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
559 output_activation_max, output_data, output_dims, gemm_context); in FullyConnected()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 568 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
609 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in FullyConnected()
623 gemmlowp::GemmContext* gemm_context) { in FullyConnected() argument
632 output_activation_max, output_data, output_dims, gemm_context); in FullyConnected()
847 gemmlowp::GemmContext* gemm_context) { in Conv() argument
904 gemm_context, filter_matrix, input_matrix, &output_matrix, filter_offset, in Conv()
920 gemmlowp::GemmContext* gemm_context) { in Conv() argument
934 im2col_data, im2col_dims, gemm_context); in Conv()
947 const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) { in Conv() argument
957 im2col_data, im2col_dims, gemm_context); in Conv()
[all …]
|