Home
last modified time | relevance | path

Searched refs:AlgorithmType (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc209 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()
Dmatmul_utils.h117 std::optional<se::blas::AlgorithmType> algorithm = std::nullopt,
Dmatmul_utils.cc428 se::blas::AlgorithmType algorithm, in DoGemmWithAlgorithm()
458 std::optional<se::blas::AlgorithmType> algorithm, in DoGemm()
489 std::optional<se::blas::AlgorithmType> algorithm, in RunGemm()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dblas.h132 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 …]
Dstream.h870 blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
885 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
924 blas::AlgorithmType algorithm, in ThenBlasGemmStridedBatchedWithAlgorithm()
Dstream_executor_pimpl.h415 std::vector<blas::AlgorithmType>* out_algorithms);
Dstream_executor_pimpl.cc366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.cc911 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()
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_fused.cc314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
Dmatmul_op_impl.h486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
/external/cronet/tot/third_party/rust/chromium_crates_io/vendor/windows-sys-0.52.0/src/Windows/Win32/Security/Cryptography/Certificates/
Dmod.rs1015 pub const XCN_BCRYPT_ASYMMETRIC_ENCRYPTION_INTERFACE: AlgorithmType = 3i32;
1016 pub const XCN_BCRYPT_CIPHER_INTERFACE: AlgorithmType = 1i32;
1017 pub const XCN_BCRYPT_HASH_INTERFACE: AlgorithmType = 2i32;
1018 pub const XCN_BCRYPT_KEY_DERIVATION_INTERFACE: AlgorithmType = 7i32;
1019 pub const XCN_BCRYPT_RNG_INTERFACE: AlgorithmType = 6i32;
1020 pub const XCN_BCRYPT_SECRET_AGREEMENT_INTERFACE: AlgorithmType = 4i32;
1021 pub const XCN_BCRYPT_SIGNATURE_INTERFACE: AlgorithmType = 5i32;
1022 pub const XCN_BCRYPT_UNKNOWN_INTERFACE: AlgorithmType = 0i32;
2338 pub type AlgorithmType = i32; typedef
/external/cronet/stable/third_party/rust/chromium_crates_io/vendor/windows-sys-0.52.0/src/Windows/Win32/Security/Cryptography/Certificates/
Dmod.rs1015 pub const XCN_BCRYPT_ASYMMETRIC_ENCRYPTION_INTERFACE: AlgorithmType = 3i32;
1016 pub const XCN_BCRYPT_CIPHER_INTERFACE: AlgorithmType = 1i32;
1017 pub const XCN_BCRYPT_HASH_INTERFACE: AlgorithmType = 2i32;
1018 pub const XCN_BCRYPT_KEY_DERIVATION_INTERFACE: AlgorithmType = 7i32;
1019 pub const XCN_BCRYPT_RNG_INTERFACE: AlgorithmType = 6i32;
1020 pub const XCN_BCRYPT_SECRET_AGREEMENT_INTERFACE: AlgorithmType = 4i32;
1021 pub const XCN_BCRYPT_SIGNATURE_INTERFACE: AlgorithmType = 5i32;
1022 pub const XCN_BCRYPT_UNKNOWN_INTERFACE: AlgorithmType = 0i32;
2338 pub type AlgorithmType = i32; typedef
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc635 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()