Searched defs:gemm_params (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| D | fully_connected.h | 79 cpu_backend_gemm::GemmParams<int32, int8> gemm_params; in FullyConnected() local
|
| D | transpose_conv.h | 85 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in TransposeConvV2() local
|
| D | conv.h | 115 gemm_params; in ConvPerChannel() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | batch_matmul.h | 107 GemmParams<float, float> gemm_params; in BatchMatMul() local
226 GemmParams<int32_t, int32_t> gemm_params; in BatchMatMul() local
365 GemmParams<int32_t, int8_t> gemm_params; in BatchMatMul() local
|
| D | sse_tensor_utils.cc | 200 GemmParams<int32, int32> gemm_params; in SseCpuBackendGemm() local
|
| D | optimized_ops.h | 359 cpu_backend_gemm::GemmParams<float, float> gemm_params; in FullyConnected() local
420 cpu_backend_gemm::GemmParams<int32, uint8> gemm_params; in FullyConnected() local
481 cpu_backend_gemm::GemmParams<int32, int16> gemm_params; in FullyConnected() local
1323 cpu_backend_gemm::GemmParams<float, float> gemm_params; in Conv() local
1511 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in HybridConvPerChannel() local
1629 cpu_backend_gemm::GemmParams<int32, uint8> gemm_params; in Conv() local
3001 cpu_backend_gemm::GemmParams<int32, int16> gemm_params; in LstmCell() local
5766 cpu_backend_gemm::GemmParams<float, float> gemm_params; in TransposeConvV2() local
5997 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in TransposeConvV2() local
|
| D | neon_tensor_utils.cc | 1003 GemmParams<int32, int32> gemm_params; in NeonCpuBackendGemm() local
|
| /external/XNNPACK/src/operators/ |
| D | convolution-nhwc.c | 775 const void* gemm_params, in setup_convolution2d_nhwc()
|