/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
H A D | blas.h | 132 typedef int64_t AlgorithmType; typedef 133 constexpr AlgorithmType kDefaultAlgorithm = -1; 134 constexpr AlgorithmType kDefaultBlasGemm = -2; 135 constexpr AlgorithmType kDefaultBlasGemv = -3; 136 constexpr AlgorithmType kNoAlgorithm = -4; 145 constexpr AlgorithmType kDefaultGemmAlgo = -1; 156 AlgorithmType algorithm() const { return algorithm_; } in algorithm() 157 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm() 163 AlgorithmType algorithm_ = kDefaultAlgorithm; 170 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() [all …]
|
H A D | stream.h | 870 blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm() 885 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm() 924 blas::AlgorithmType algorithm, in ThenBlasGemmStridedBatchedWithAlgorithm()
|
H A D | stream_executor_pimpl.h | 415 std::vector<blas::AlgorithmType>* out_algorithms);
|
H A D | stream_executor_pimpl.cc | 366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gemm_algorithm_picker.cc | 209 StatusOr<std::optional<se::blas::AlgorithmType>> DoGemmAutotune( in DoGemmAutotune() 227 std::optional<se::blas::AlgorithmType>>(); in DoGemmAutotune() 267 std::optional<se::blas::AlgorithmType> best_algorithm; in DoGemmAutotune() 308 std::vector<se::blas::AlgorithmType> algorithms; in DoGemmAutotune() 312 GetBestAlgorithm<se::blas::AlgorithmType>( in DoGemmAutotune() 315 [&](const se::blas::AlgorithmType& algorithm) in DoGemmAutotune() 350 TF_ASSIGN_OR_RETURN(std::optional<se::blas::AlgorithmType> gemm_algorithm, in RunOnInstruction()
|
H A D | matmul_utils.cc | 428 se::blas::AlgorithmType algorithm, in DoGemmWithAlgorithm() 458 std::optional<se::blas::AlgorithmType> algorithm, in DoGemm() 489 std::optional<se::blas::AlgorithmType> algorithm, in RunGemm()
|
H A D | matmul_utils.h | 117 std::optional<se::blas::AlgorithmType> algorithm = std::nullopt,
|
/aosp_15_r20/prebuilts/gcc/linux-x86/host/x86_64-w64-mingw32-4.8/x86_64-w64-mingw32/include/ |
D | certenroll.h | 34 typedef enum AlgorithmType { enum 42 } AlgorithmType; typedef
|
D | bdaiface.h | 6240 DWORD *AlgorithmType, 6248 DWORD *AlgorithmType, 6315 DWORD *AlgorithmType, 6324 DWORD *AlgorithmType, 6377 …tIndex,MethodID,AlgorithmType,SamplingRate,BitDepth,NumChannels) (This)->lpVtbl->EnumAudioCapabili… 6378 …AlgorithmType,VerticalSize,HorizontalSize,AspectRatio,FrameRateCode,ProgressiveSequence) (This)->l… 6396 …ioCapability(IBDA_Encoder* This,DWORD FmtIndex,DWORD *MethodID,DWORD *AlgorithmType,DWORD *Samplin… 6397 …return This->lpVtbl->EnumAudioCapability(This,FmtIndex,MethodID,AlgorithmType,SamplingRate,BitDept… 6399 …eoCapability(IBDA_Encoder* This,DWORD FmtIndex,DWORD *MethodID,DWORD *AlgorithmType,DWORD *Vertica… 6400 …return This->lpVtbl->EnumVideoCapability(This,FmtIndex,MethodID,AlgorithmType,VerticalSize,Horizon… [all …]
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
H A D | cuda_blas.cc | 911 static bool UsesTensorOps(blas::AlgorithmType algo) { in DoBlasInternalImpl() 921 Stream *stream, blas::AlgorithmType algorithm, blas::DataType type_a, in DoBlasInternalImpl() 999 GpuTimer *timer, blas::AlgorithmType algorithm, in DoBlasInternalImpl() 1021 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() 1051 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl() 1098 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()
|
/aosp_15_r20/external/python/cpython2/Lib/plat-irix5/ |
D | CL_old.py | 104 def AlgorithmType(scheme): function
|
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
H A D | rocm_blas.cc | 635 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm() 647 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmStridedBatchedWithAlgorithm() 653 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | matmul_op_fused.cc | 314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
|
H A D | matmul_op_impl.h | 486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
|
/aosp_15_r20/external/python/cpython2/Modules/ |
D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|