Home
last modified time | relevance | path

Searched refs:AlgorithmDesc (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_algorithm_denylist_test.cc56 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(0, false), list[0]); in TEST_F()
57 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(0, true), list[1]); in TEST_F()
58 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(1, false), list[2]); in TEST_F()
59 EXPECT_EQ(stream_executor::dnn::AlgorithmDesc(1, true), list[3]); in TEST_F()
Dgpu_conv_algorithm_picker.cc51 using se::dnn::AlgorithmDesc;
102 std::vector<AlgorithmDesc> GetAlgorithms(CudnnConvKind kind, in GetAlgorithms()
104 std::vector<AlgorithmDesc> algorithms; in GetAlgorithms()
152 string AlgorithmToString(const AlgorithmDesc& algo) { in AlgorithmToString()
426 AlgorithmDesc first_algorithm; in PickBestAlgorithmNoCacheCuda()
445 absl::Span<const AlgorithmDesc> disabled_algos = GetDisabledConvAlgorithms( in PickBestAlgorithmNoCacheCuda()
451 for (const AlgorithmDesc& alg : GetAlgorithms(kind, stream_exec_)) { in PickBestAlgorithmNoCacheCuda()
Dhlo_algorithm_denylist.h27 absl::Span<const stream_executor::dnn::AlgorithmDesc> GetDisabledConvAlgorithms(
Dgpu_conv_runner.h39 absl::optional<se::dnn::AlgorithmDesc> algo_override;
Dgpu_conv_runner.cc275 se::dnn::AlgorithmDesc(backend_config.algorithm(), in GetGpuConvConfig()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h742 class AlgorithmDesc {
745 AlgorithmDesc() : AlgorithmDesc(0, false) {} in AlgorithmDesc() function
746 AlgorithmDesc(Index a, bool use_tensor_ops) { in AlgorithmDesc() function
755 bool operator==(const AlgorithmDesc& other) const {
781 AlgorithmDesc algorithm() const { return *algorithm_; } in algorithm()
782 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; } in set_algorithm()
791 absl::optional<AlgorithmDesc> algorithm_;
817 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
818 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size) in AlgorithmConfig()
820 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig()
[all …]
Ddnn.cc31 uint64 AlgorithmDesc::hash() const { in hash()
36 std::string AlgorithmDesc::ToString() const { in ToString()
46 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
64 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
70 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
76 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream_executor_pimpl.h366 std::vector<dnn::AlgorithmDesc> *out_algorithms);
382 bool GetRnnAlgorithms(std::vector<dnn::AlgorithmDesc> *out_algorithms);
387 std::vector<dnn::AlgorithmDesc> *out_algorithms);
393 std::vector<dnn::AlgorithmDesc> *out_algorithms);
Dstream_executor_pimpl.cc266 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
297 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
307 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
320 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream.h72 class AlgorithmDesc; variable
337 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm()
448 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardDataWithAlgorithm()
480 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardFilterWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc207 cudnnConvolutionFwdAlgo_t ToConvForwardAlgo(dnn::AlgorithmDesc algorithm) { in ToConvForwardAlgo()
227 dnn::AlgorithmDesc algorithm) { in ToConvBackwardDataAlgo()
246 dnn::AlgorithmDesc algorithm) { in ToConvBackwardFilterAlgo()
272 cudnnRNNAlgo_t ToCudnnRNNAlgo(absl::optional<dnn::AlgorithmDesc> algorithm) { in ToCudnnRNNAlgo()
2539 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2583 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2629 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2670 absl::optional<dnn::AlgorithmDesc> desc) { in UseTensorOps()
2687 port::StatusOr<dnn::AlgorithmDesc> GetCudnnConvolutionForwardAlgorithm( in GetCudnnConvolutionForwardAlgorithm()
2695 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm()
[all …]
Dcuda_dnn.h205 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
208 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
212 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
216 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
277 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
703 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc63 using dnn::AlgorithmDesc;
508 {static_cast<AlgorithmDesc::Index>(solution.solution_id), false}); in GetProfileResultFromConvSolution()
520 {static_cast<AlgorithmDesc::Index>(algorithm.fwd_algo), false}); in GetProfileResultFromConvAlgoPerf()
524 {static_cast<AlgorithmDesc::Index>(algorithm.bwd_data_algo), false}); in GetProfileResultFromConvAlgoPerf()
528 {static_cast<AlgorithmDesc::Index>(algorithm.bwd_weights_algo), in GetProfileResultFromConvAlgoPerf()
2851 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2853 absl::optional<dnn::AlgorithmDesc> input_algo_desc = in DoPrepareForConvolution()
2946 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3065 dnn::AlgorithmDesc algotype(algorithm_desc.algo_id(), false); in DoConvolve()
3085 std::vector<dnn::AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
[all …]
Drocm_dnn.h232 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
246 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
250 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
254 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
315 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
842 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc125 using se::dnn::AlgorithmDesc;
530 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& in operator ()()
549 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& lhs, in operator ()()
550 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& rhs) in operator ()()
1076 std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>,
1498 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm()
1645 std::vector<AlgorithmDesc> algorithms; in MaybeAutoTune()
2052 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm()
Dconv_grad_input_ops.cc73 using se::dnn::AlgorithmDesc; in operator ()()
402 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_ops_3d.cc496 using se::dnn::AlgorithmDesc; in launch()
516 std::vector<AlgorithmDesc> algorithms; in launch()
Dconv_grad_ops_3d.cc1416 using se::dnn::AlgorithmDesc; in Compute()
1434 std::vector<AlgorithmDesc> algorithms; in Compute()
1913 using se::dnn::AlgorithmDesc; in Compute()
1925 std::vector<AlgorithmDesc> algorithms; in Compute()
Dconv_grad_filter_ops.cc657 using se::dnn::AlgorithmDesc; in operator ()()
996 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_ops.cc653 using se::dnn::AlgorithmDesc; in operator ()()
995 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_ops_fused_impl.h376 std::vector<se::dnn::AlgorithmDesc> algorithms;