/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | hlo_algorithm_denylist_test.cc | 56 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()
|
D | gpu_conv_algorithm_picker.cc | 51 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()
|
D | hlo_algorithm_denylist.h | 27 absl::Span<const stream_executor::dnn::AlgorithmDesc> GetDisabledConvAlgorithms(
|
D | gpu_conv_runner.h | 39 absl::optional<se::dnn::AlgorithmDesc> algo_override;
|
D | gpu_conv_runner.cc | 275 se::dnn::AlgorithmDesc(backend_config.algorithm(), in GetGpuConvConfig()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 742 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 …]
|
D | dnn.cc | 31 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()
|
D | stream_executor_pimpl.h | 366 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);
|
D | stream_executor_pimpl.cc | 266 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()
|
D | stream.h | 72 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/ |
D | cuda_dnn.cc | 207 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 …]
|
D | cuda_dnn.h | 205 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/ |
D | rocm_dnn.cc | 63 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 …]
|
D | rocm_dnn.h | 232 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/ |
D | cudnn_rnn_ops.cc | 125 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()
|
D | conv_grad_input_ops.cc | 73 using se::dnn::AlgorithmDesc; in operator ()() 402 std::vector<AlgorithmDesc> algorithms; in operator ()()
|
D | conv_ops_3d.cc | 496 using se::dnn::AlgorithmDesc; in launch() 516 std::vector<AlgorithmDesc> algorithms; in launch()
|
D | conv_grad_ops_3d.cc | 1416 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()
|
D | conv_grad_filter_ops.cc | 657 using se::dnn::AlgorithmDesc; in operator ()() 996 std::vector<AlgorithmDesc> algorithms; in operator ()()
|
D | conv_ops.cc | 653 using se::dnn::AlgorithmDesc; in operator ()() 995 std::vector<AlgorithmDesc> algorithms; in operator ()()
|
D | conv_ops_fused_impl.h | 376 std::vector<se::dnn::AlgorithmDesc> algorithms;
|