Home
last modified time | relevance | path

Searched refs:DnnSupport (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc44 bool DnnSupport::GetConvolveAlgorithms( in GetConvolveAlgorithms()
50 bool DnnSupport::GetMIOpenConvolveAlgorithms( in GetMIOpenConvolveAlgorithms()
64 bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { in GetRnnAlgorithms()
68 bool DnnSupport::GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms()
74 bool DnnSupport::GetConvolveBackwardFilterAlgorithms( in GetConvolveBackwardFilterAlgorithms()
613 bool DnnSupport::IsStatusOk(const port::Status& status, bool report_error) { in IsStatusOk()
623 port::Status DnnSupport::DoCtcLoss( in DoCtcLoss()
Dstream_executor_pimpl.cc267 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms()
286 dnn::DnnSupport *dnn_support = AsDnn(); in GetMIOpenConvolveAlgorithms()
298 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms()
308 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms()
321 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms()
372 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnDescriptor()
387 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()
401 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor()
415 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnStateTensorDescriptor()
424 dnn::DnnSupport *StreamExecutor::AsDnn() { in AsDnn()
Dstream.cc358 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()
382 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()
409 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationForward()
434 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenBatchNormalizationBackward()
466 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()
497 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()
529 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()
560 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()
591 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in FusedConvolveWithAlgorithm()
635 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ThenConvolveQuantized()
[all …]
Dplugin_registry.h53 typedef dnn::DnnSupport* (*DnnFactory)(internal::StreamExecutorInterface*);
Dstream_executor_pimpl.h494 dnn::DnnSupport *AsDnn();
688 std::unique_ptr<dnn::DnnSupport> dnn_ TF_GUARDED_BY(mu_);
Dstream_executor_internal.h357 virtual dnn::DnnSupport *CreateDnn() { return nullptr; } in CreateDnn()
Ddnn.h968 class DnnSupport {
970 DnnSupport() {} in DnnSupport() function
971 virtual ~DnnSupport() {} in ~DnnSupport()
2706 SE_DISALLOW_COPY_AND_ASSIGN(DnnSupport);
Dstream.h338 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveWithAlgorithm()
449 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardDataWithAlgorithm()
481 if (dnn::DnnSupport *dnn = parent_->AsDnn()) { in ConvolveBackwardFilterWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h155 dnn::DnnSupport *CreateDnn() override { return nullptr; } in CreateDnn()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h220 dnn::DnnSupport* CreateDnn() override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc668 dnn::DnnSupport* GpuExecutor::CreateDnn() { in CreateDnn()
Drocm_dnn.h77 class MIOpenSupport : public dnn::DnnSupport {
Drocm_dnn.cc5086 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_miopen()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc760 dnn::DnnSupport* GpuExecutor::CreateDnn() { in CreateDnn()
Dcuda_dnn.h43 class CudnnSupport : public dnn::DnnSupport {
Dcuda_dnn.cc4655 [](internal::StreamExecutorInterface* parent) -> dnn::DnnSupport* { in initialize_cudnn()