Home
last modified time | relevance | path

Searched refs:cpu_backend_gemm (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h64 cpu_backend_gemm::MatrixParams<int8> lhs_params; in FullyConnected()
67 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in FullyConnected()
69 cpu_backend_gemm::MatrixParams<int8> rhs_params; in FullyConnected()
72 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
74 cpu_backend_gemm::MatrixParams<int8> dst_params; in FullyConnected()
77 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
79 cpu_backend_gemm::GemmParams<int32, int8> gemm_params; in FullyConnected()
85 cpu_backend_gemm::Gemm(lhs_params, filter_data, rhs_params, input_data, in FullyConnected()
Dconv.h97 cpu_backend_gemm::MatrixParams<int8> lhs_params; in ConvPerChannel()
100 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in ConvPerChannel()
102 cpu_backend_gemm::MatrixParams<int8> rhs_params; in ConvPerChannel()
105 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in ConvPerChannel()
107 cpu_backend_gemm::MatrixParams<int8> dst_params; in ConvPerChannel()
110 dst_params.order = cpu_backend_gemm::Order::kColMajor; in ConvPerChannel()
112 cpu_backend_gemm::GemmParams< in ConvPerChannel()
114 cpu_backend_gemm::QuantizationFlavor::kIntegerWithPerRowMultiplier> in ConvPerChannel()
121 cpu_backend_gemm::Gemm(lhs_params, filter_data, rhs_params, gemm_input_data, in ConvPerChannel()
Dtranspose_conv.h64 cpu_backend_gemm::MatrixParams<int8_t> lhs_params; in TransposeConvV2()
65 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in TransposeConvV2()
74 cpu_backend_gemm::MatrixParams<int8_t> rhs_params; in TransposeConvV2()
75 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in TransposeConvV2()
80 cpu_backend_gemm::MatrixParams<int32_t> dst_params; in TransposeConvV2()
81 dst_params.order = cpu_backend_gemm::Order::kColMajor; in TransposeConvV2()
85 cpu_backend_gemm::GemmParams<int32_t, int32_t> gemm_params; in TransposeConvV2()
86 cpu_backend_gemm::Gemm(lhs_params, hwoi_ordered_filter_data, rhs_params, in TransposeConvV2()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h32 using ::tflite::cpu_backend_gemm::Gemm; in BatchMatMul()
33 using ::tflite::cpu_backend_gemm::GemmParams; in BatchMatMul()
34 using ::tflite::cpu_backend_gemm::MatrixParams; in BatchMatMul()
81 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in BatchMatMul()
86 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
91 dst_params.order = cpu_backend_gemm::Order::kColMajor; in BatchMatMul()
108 cpu_backend_gemm::Gemm(lhs_params, lhs_ptr2, rhs_params, rhs_ptr2, in BatchMatMul()
122 using ::tflite::cpu_backend_gemm::Gemm; in BatchMatMul()
123 using ::tflite::cpu_backend_gemm::GemmParams; in BatchMatMul()
124 using ::tflite::cpu_backend_gemm::MatrixParams; in BatchMatMul()
[all …]
Dsse_tensor_utils.cc180 using ::tflite::cpu_backend_gemm::Gemm; in SseCpuBackendGemm()
181 using ::tflite::cpu_backend_gemm::GemmParams; in SseCpuBackendGemm()
182 using ::tflite::cpu_backend_gemm::MatrixParams; in SseCpuBackendGemm()
185 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in SseCpuBackendGemm()
188 lhs_params.cache_policy = cpu_backend_gemm::CachePolicy::kCacheIfLargeSpeedup; in SseCpuBackendGemm()
191 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in SseCpuBackendGemm()
196 dst_params.order = cpu_backend_gemm::Order::kColMajor; in SseCpuBackendGemm()
204 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in SseCpuBackendGemm()
Doptimized_ops.h341 cpu_backend_gemm::MatrixParams<float> rhs_params; in FullyConnected()
342 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
346 cpu_backend_gemm::DefaultCachePolicy(params.rhs_cacheable); in FullyConnected()
348 cpu_backend_gemm::MatrixParams<float> lhs_params; in FullyConnected()
349 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in FullyConnected()
353 cpu_backend_gemm::DefaultCachePolicy(params.lhs_cacheable); in FullyConnected()
354 cpu_backend_gemm::MatrixParams<float> dst_params; in FullyConnected()
355 dst_params.order = cpu_backend_gemm::Order::kColMajor; in FullyConnected()
359 cpu_backend_gemm::GemmParams<float, float> gemm_params; in FullyConnected()
363 cpu_backend_gemm::Gemm(lhs_params, weights_data, rhs_params, input_data, in FullyConnected()
[all …]
Dneon_tensor_utils.cc983 using ::tflite::cpu_backend_gemm::Gemm; in NeonCpuBackendGemm()
984 using ::tflite::cpu_backend_gemm::GemmParams; in NeonCpuBackendGemm()
985 using ::tflite::cpu_backend_gemm::MatrixParams; in NeonCpuBackendGemm()
988 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in NeonCpuBackendGemm()
991 lhs_params.cache_policy = cpu_backend_gemm::CachePolicy::kCacheIfLargeSpeedup; in NeonCpuBackendGemm()
994 rhs_params.order = cpu_backend_gemm::Order::kColMajor; in NeonCpuBackendGemm()
999 dst_params.order = cpu_backend_gemm::Order::kColMajor; in NeonCpuBackendGemm()
1007 cpu_backend_gemm::Gemm(lhs_params, input_to_gate_weights, rhs_params, input, in NeonCpuBackendGemm()
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_test.cc43 using cpu_backend_gemm::Gemm;
44 using cpu_backend_gemm::GemmParams;
45 using cpu_backend_gemm::MatrixParams;
46 using cpu_backend_gemm::QuantizationFlavor;
356 cpu_backend_gemm::detail::MakeRuyMatrix(lhs_params, lhs_data, &ruy_lhs); in ReferenceGemm()
357 cpu_backend_gemm::detail::MakeRuyMatrix(rhs_params, rhs_data, &ruy_rhs); in ReferenceGemm()
358 cpu_backend_gemm::detail::MakeRuyMatrix(dst_params, dst_data, &ruy_dst); in ReferenceGemm()
361 cpu_backend_gemm::detail::MakeRuyMulParams(params, &ruy_mul_params); in ReferenceGemm()
393 return random_engine() % 2 ? cpu_backend_gemm::Order::kRowMajor in TestSomeGemm()
394 : cpu_backend_gemm::Order::kColMajor; in TestSomeGemm()
[all …]
Dcpu_backend_gemm_eigen.h25 namespace cpu_backend_gemm {
Dcpu_backend_gemm_eigen.cc32 namespace cpu_backend_gemm { namespace
Dcpu_backend_gemm_x86.h32 namespace cpu_backend_gemm {
Dcpu_backend_gemm_params.h27 namespace cpu_backend_gemm {
Dcpu_backend_gemm.h35 namespace cpu_backend_gemm {
Dcpu_backend_gemm_ruy.h27 namespace cpu_backend_gemm {
Dcpu_backend_gemm_gemmlowp.h33 namespace cpu_backend_gemm {
DBUILD324 # arithmetic backends. Include as a dependency of cpu_backend_gemm to start.
407 name = "cpu_backend_gemm",
416 "cpu_backend_gemm.h",
431 # cpu_backend_gemm.h about why ruy is the generic path.
451 ":cpu_backend_gemm",
648 ":cpu_backend_gemm",
Dcpu_backend_gemm_custom_gemv.h50 namespace cpu_backend_gemm {
/external/tensorflow/tensorflow/lite/kernels/internal/
DBUILD279 "//tensorflow/lite/kernels:cpu_backend_gemm",
325 "//tensorflow/lite/kernels:cpu_backend_gemm",
740 "//tensorflow/lite/kernels:cpu_backend_gemm",
763 "//tensorflow/lite/kernels:cpu_backend_gemm",