Home
last modified time | relevance | path

Searched defs:gemm (Results 1 – 25 of 60) sorted by relevance

123

/aosp_15_r20/external/executorch/kernels/optimized/blas/
H A DCPUBlas.cpp75 void gemm( in gemm() function
115 void gemm( in gemm() function
155 void gemm( in gemm() function
178 void gemm( in gemm() function
H A DCPUBlas.h119 void gemm( in gemm() function
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCUDABlas.cpp1094 static tunable::GemmTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::T> gemm{}; in gemm_tunable() local
1098 static tunable::GemmTunableOp<DType, tunable::BlasOp::T, tunable::BlasOp::N> gemm{}; in gemm_tunable() local
1102 static tunable::GemmTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::T> gemm{}; in gemm_tunable() local
1106 static tunable::GemmTunableOp<DType, tunable::BlasOp::N, tunable::BlasOp::N> gemm{}; in gemm_tunable() local
1115 void gemm<double>(CUDABLAS_GEMM_ARGTYPES(double)) { in gemm() function
1126 void gemm<float>(CUDABLAS_GEMM_ARGTYPES(float)) { in gemm() function
1137 void gemm<c10::complex<double>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<double>)) { in gemm() function
1148 void gemm<c10::complex<float>>(CUDABLAS_GEMM_ARGTYPES(c10::complex<float>)) { in gemm() function
1159 void gemm<at::Half>(CUDABLAS_GEMM_ARGTYPES(at::Half)) { in gemm() function
1170 void gemm<at::BFloat16>(CUDABLAS_GEMM_ARGTYPES(at::BFloat16)) { in gemm() function
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGEMMFixture.h96 FunctionType gemm; in compute_target() local
280 GEMMOperatorType gemm; in compute_target() local
403 GEMMOperatorType gemm; in compute_target() local
544 GEMMOperatorType gemm; in compute_target() local
699 GEMMOperatorType gemm; in compute_target() local
866 GEMMOperatorType gemm; in compute_target() local
1075 GEMMOperatorType gemm; in compute_target() local
1304 GEMMOperatorType gemm; in compute_target() local
1481 GEMMOperatorType gemm; in compute_target() local
1670 GEMMOperatorType gemm; in compute_target() local
[all …]
H A DGEMMLowpFixture.h1007 GEMMFunctionType gemm; in compute_target() local
1162 GEMMFunctionType gemm; in compute_target() local
1317 GEMMFunctionType gemm; in compute_target() local
1496 GEMMFunctionType gemm; in compute_target() local
1671 GEMMFunctionType gemm; in compute_target() local
1828 GEMMFunctionType gemm; in compute_target() local
1945 GEMMFunctionType gemm; in compute_target() local
2043 GEMMFunctionType gemm; in compute_target() local
H A DRNNLayerFixture.h137 …SimpleTensor<T> gemm = reference::gemm(hidden_state, recurrent_weights, out_w, 1.f, 0.f… in compute_reference() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DCPUBlas.cpp127 void gemm( in gemm() function
170 void gemm( in gemm() function
218 void gemm( in gemm() function
261 void gemm( in gemm() function
304 void gemm( in gemm() function
344 void gemm( in gemm() function
363 void gemm( in gemm() function
414 void gemm( in gemm() function
450 void gemm( in gemm() function
H A DCPUBlas.h33 void gemm( in gemm() function
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DGEMM.cpp100 cpu::kernels::CpuGemmMatrixMultiplyKernel gemm; in validate_gemm_zero_padding() local
119 auto gemm = std::make_unique<cpu::CpuGemm>(); in TEST_SUITE() local
172 auto gemm = std::make_unique<NEGEMM>(); in TEST_CASE() local
H A DGEMMLowp.cpp124 auto gemm = std::make_unique<cpu::CpuGemmLowpMatrixMultiplyCore>(); in TEST_CASE() local
185 auto gemm = std::make_unique<NEGEMMLowpMatrixMultiplyCore>(); in TEST_CASE() local
/aosp_15_r20/external/gemmlowp/test/
H A Dbenchmark.cc77 for (auto gemm : gemms) { in time_for_gemms() local
144 for (auto gemm : gemms) { in gflops_for_gemms() local
184 for (auto gemm : benchmark_gemms) { in benchmark() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBlas.cpp236 …:GemmAndBiasTunableOp<scalar_t, at::cuda::tunable::BlasOp::T, at::cuda::tunable::BlasOp::T> gemm{}; in launchTunableGemmAndBias() local
240 …:GemmAndBiasTunableOp<scalar_t, at::cuda::tunable::BlasOp::T, at::cuda::tunable::BlasOp::N> gemm{}; in launchTunableGemmAndBias() local
244 …:GemmAndBiasTunableOp<scalar_t, at::cuda::tunable::BlasOp::N, at::cuda::tunable::BlasOp::T> gemm{}; in launchTunableGemmAndBias() local
248 …:GemmAndBiasTunableOp<scalar_t, at::cuda::tunable::BlasOp::N, at::cuda::tunable::BlasOp::N> gemm{}; in launchTunableGemmAndBias() local
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dparams.h88 struct xnn_hmp_gemm_ukernel gemm[XNN_MAX_MR]; member
153 struct xnn_hmp_gemm_codegen gemm; member
499 struct gemm_parameters gemm; member
503 struct gemm_parameters gemm; member
511 struct gemm_parameters gemm; member
542 struct gemm_parameters gemm; member
593 struct gemm_parameters gemm; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_rewriter.cc47 Status SetName(HloModule *module, HloInstruction *gemm) { in SetName()
258 HloInstruction *gemm) { in FuseMatrixBiasAdd()
H A Dgemm_algorithm_picker.cc95 const HloInstruction& gemm, const AutotuneConfig& autotune_config, in GetBestAlgorithm()
210 const HloInstruction* gemm, const GemmBackendConfig& gemm_config, in DoGemmAutotune()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_mkldnn_contraction_kernel_test.cc139 Scalar gemm = mkldnn_result(i); in TEST() local
210 Scalar gemm = res(i, j); in TEST() local
/aosp_15_r20/external/eigen/bench/perf_monitoring/
H A Dgemm.cpp4 void gemm(const Mat &A, const Mat &B, Mat &C) in gemm() function
/aosp_15_r20/external/XNNPACK/test/
H A Dgemm-microkernel-tester.cc27 xnn_qu8_gemm_minmax_ukernel_function gemm, in Test()
239 xnn_qc8_gemm_minmax_ukernel_function gemm, in Test()
494 xnn_qs8_gemm_minmax_ukernel_function gemm, in Test()
1999 …xnn_qc8_gemm_minmax_ukernel_function gemm = reinterpret_cast<xnn_qc8_gemm_minmax_ukernel_function>… in Test() local
2254 …xnn_qs8_gemm_minmax_ukernel_function gemm = reinterpret_cast<xnn_qs8_gemm_minmax_ukernel_function … in Test() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/gemm/native/
H A DClGemmDefaultConfigNativeMidgard.cpp39 namespace gemm namespace
H A DClGemmDefaultConfigNativeValhall.cpp39 namespace gemm namespace
H A DClGemmDefaultConfigNativeBifrost.cpp39 namespace gemm namespace
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dautotuning.proto83 GemmKey gemm = 6; field
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DGEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp159 CLGEMMLowpMatrixMultiplyReshapedOnlyRHS gemm; in validate_configuration() local
/aosp_15_r20/external/ComputeLibrary/examples/gemm_tuner/
H A Dcl_gemm_native.cpp245 CLGEMMMatrixMultiplyNative gemm{}; member in CLGEMMMatrixMultiplyNativeExample
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/gemm/
H A DClGemmHelpers.cpp39 namespace gemm namespace

123