Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h738 class AlgorithmDesc {
741 AlgorithmDesc() : AlgorithmDesc(0, false) {}
742 AlgorithmDesc(Index a, bool use_tensor_ops) {
751 bool operator==(const AlgorithmDesc& other) const {
775 AlgorithmDesc algorithm() const { return *algorithm_; }
776 void set_algorithm(AlgorithmDesc val) { algorithm_ = val; }
785 absl::optional<AlgorithmDesc> algorithm_;
801 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {}
802 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch)
804 absl::optional<AlgorithmDesc> algorithm() const { return algorithm_; }
[all …]
Ddnn.cc25 uint64 AlgorithmDesc::hash() const { in hash()
31 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveAlgorithms()
35 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
41 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardDataAlgorithms()
47 std::vector<AlgorithmDesc>* out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
225 AlgorithmDesc::Index algo_id = -1; in ToString()
229 AlgorithmDesc::Index algo_id_no_scratch = -1; in ToString()
Dstream_executor_pimpl.h386 std::vector<dnn::AlgorithmDesc> *out_algorithms);
389 bool GetRnnAlgorithms(std::vector<dnn::AlgorithmDesc> *out_algorithms);
394 std::vector<dnn::AlgorithmDesc> *out_algorithms);
400 std::vector<dnn::AlgorithmDesc> *out_algorithms);
Dstream_executor_pimpl.cc325 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms()
337 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms()
347 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms()
360 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms()
Dstream.cc611 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm()
653 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm()
695 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm()
840 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm()
885 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm()
930 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm()
975 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm()
1020 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm()
1065 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm()
Dstream.h71 class AlgorithmDesc; variable
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_algorithm_picker.cc41 using se::dnn::AlgorithmDesc;
44 std::vector<AlgorithmDesc> GetAlgorithms(CudnnConvKind kind, in GetAlgorithms()
46 std::vector<AlgorithmDesc> algorithms; in GetAlgorithms()
66 string AlgorithmToString(const AlgorithmDesc& algo) { in AlgorithmToString()
232 optional<AlgorithmDesc> first_algorithm; in PickBestAlgorithm()
235 for (const AlgorithmDesc& alg : GetAlgorithms(kind, stream_exec_)) { in PickBestAlgorithm()
Dcudnn_conv_runner.h36 absl::optional<se::dnn::AlgorithmDesc> algo_override;
Dcudnn_conv_runner.cc337 params.algorithm = se::dnn::AlgorithmConfig(se::dnn::AlgorithmDesc( in GetCudnnConvParams()
/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()
274 cudnnRNNAlgo_t ToCudnnRNNAlgo(absl::optional<dnn::AlgorithmDesc> algorithm) { in ToCudnnRNNAlgo()
2214 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2257 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2302 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2341 port::StatusOr<dnn::AlgorithmDesc> GetCudnnConvolutionForwardAlgorithm( in GetCudnnConvolutionForwardAlgorithm()
2348 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in GetCudnnConvolutionForwardAlgorithm()
2361 algo_desc = dnn::AlgorithmDesc(algo, /*use_tensor_ops=*/true); in GetCudnnConvolutionForwardAlgorithm()
[all …]
Dcuda_dnn.h204 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
207 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
211 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
215 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
270 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
660 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc393 miopenConvFwdAlgorithm_t ToConvForwardAlgo(dnn::AlgorithmDesc algorithm) { in ToConvForwardAlgo()
408 dnn::AlgorithmDesc algorithm) { in ToConvBackwardDataAlgo()
425 dnn::AlgorithmDesc algorithm) { in ToConvBackwardFilterAlgo()
2599 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2616 absl::optional<dnn::AlgorithmDesc> algo_desc = algorithm_config.algorithm(); in DoPrepareForConvolution()
2681 *algorithm_desc = dnn::AlgorithmDesc(preference.fwd_algo, false); in DoPrepareForConvolution()
2699 *algorithm_desc = dnn::AlgorithmDesc(preference.bwd_data_algo, false); in DoPrepareForConvolution()
2718 dnn::AlgorithmDesc(preference.bwd_weights_algo, false); in DoPrepareForConvolution()
2816 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
2930 dnn::AlgorithmDesc algotype(algorithm_desc.algo_id(), false); in DoConvolve()
[all …]
Drocm_dnn.h195 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
198 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
202 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
206 std::vector<dnn::AlgorithmDesc>* out_algorithms) override;
261 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
767 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc125 using se::dnn::AlgorithmDesc;
526 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& in operator ()()
545 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& lhs, in operator ()()
546 const std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>& rhs) in operator ()()
996 std::pair<CudnnRnnModelShapes, absl::optional<AlgorithmDesc>>,
1313 AlgorithmDesc algo_desc(debug_cudnn_rnn_algo_, debug_use_tensor_ops_); in ComputeAndReturnAlgorithm()
1459 std::vector<AlgorithmDesc> algorithms; in MaybeAutoTune()
1851 const AlgorithmDesc algo_desc(host_reserved_int8(0), host_reserved_int8(1)); in GetAlgorithm()
Dconv_ops_3d.cc432 using se::dnn::AlgorithmDesc; in launch()
439 std::vector<AlgorithmDesc> algorithms; in launch()
Dconv_grad_ops_3d.cc1355 using se::dnn::AlgorithmDesc; in Compute()
1360 std::vector<AlgorithmDesc> algorithms; in Compute()
1761 using se::dnn::AlgorithmDesc; in Compute()
1766 std::vector<AlgorithmDesc> algorithms; in Compute()
Dconv_ops.cc585 using se::dnn::AlgorithmDesc; in operator ()()
850 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_grad_filter_ops.cc550 using se::dnn::AlgorithmDesc; in operator ()()
842 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_grad_input_ops.cc687 using se::dnn::AlgorithmDesc; in operator ()()
954 std::vector<AlgorithmDesc> algorithms; in operator ()()
Dconv_ops_fused_impl.h514 std::vector<se::dnn::AlgorithmDesc> algorithms;
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc695 std::vector<dnn::AlgorithmDesc> algorithms; in launch()
708 [](dnn::AlgorithmDesc alg) { in launch()