Home
last modified time | relevance | path

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

123

/external/executorch/kernels/optimized/blas/
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
DCPUBlas.h119 void gemm( in gemm() function
/external/ComputeLibrary/tests/validation/NEON/
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
DGEMMLowp.cpp124 auto gemm = std::make_unique<cpu::CpuGemmLowpMatrixMultiplyCore>(); in TEST_CASE() local
185 auto gemm = std::make_unique<NEGEMMLowpMatrixMultiplyCore>(); in TEST_CASE() local
/external/eigen/bench/perf_monitoring/
Dgemm.cpp4 void gemm(const Mat &A, const Mat &B, Mat &C) in gemm() function
/external/gemmlowp/test/
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
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_rewriter.cc47 Status SetName(HloModule *module, HloInstruction *gemm) { in SetName()
258 HloInstruction *gemm) { in FuseMatrixBiasAdd()
Dgemm_algorithm_picker.cc95 const HloInstruction& gemm, const AutotuneConfig& autotune_config, in GetBestAlgorithm()
210 const HloInstruction* gemm, const GemmBackendConfig& gemm_config, in DoGemmAutotune()
/external/pytorch/aten/src/ATen/cuda/
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
DCUDABlas.h50 inline void gemm(CUDABLAS_GEMM_ARGTYPES(Dtype)) { in gemm() function
/external/ComputeLibrary/src/gpu/cl/kernels/gemm/native/
DClGemmDefaultConfigNativeMidgard.cpp39 namespace gemm namespace
DClGemmDefaultConfigNativeValhall.cpp39 namespace gemm namespace
DClGemmDefaultConfigNativeBifrost.cpp39 namespace gemm namespace
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc139 Scalar gemm = mkldnn_result(i); in TEST() local
210 Scalar gemm = res(i, j); in TEST() local
/external/ComputeLibrary/tests/validation/fixtures/
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 …]
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
DRNNLayerFixture.h137 …SimpleTensor<T> gemm = reference::gemm(hidden_state, recurrent_weights, out_w, 1.f, 0.f… in compute_reference() local
/external/pytorch/aten/src/ATen/native/
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
DCPUBlas.h33 void gemm( in gemm() function
/external/pytorch/aten/src/ATen/native/cuda/
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
/external/tensorflow/tensorflow/core/protobuf/
Dautotuning.proto83 GemmKey gemm = 6; field
/external/ComputeLibrary/tests/validation/CL/
DGEMMLowpMatrixMultiplyReshapedOnlyRHS.cpp159 CLGEMMLowpMatrixMultiplyReshapedOnlyRHS gemm; in validate_configuration() local
/external/ComputeLibrary/examples/gemm_tuner/
Dcl_gemm_native.cpp245 CLGEMMMatrixMultiplyNative gemm{}; member in CLGEMMMatrixMultiplyNativeExample
/external/ComputeLibrary/src/gpu/cl/kernels/gemm/reshaped/
DClGemmDefaultConfigReshapedValhall.cpp39 namespace gemm namespace
/external/ComputeLibrary/src/gpu/cl/kernels/gemm/
DClGemmHelpers.cpp39 namespace gemm namespace

123