/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 43 class CudnnSupport : public dnn::DnnSupport { 48 port::StatusOr<perftools::gputools::dnn::VersionInfo> GetVersion() override; 50 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor( 52 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, 53 dnn::RnnMode rnn_mode, dnn::DataType data_type, 54 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, 57 port::StatusOr<std::unique_ptr<dnn::RnnSequenceTensorDescriptor>> 60 dnn::DataType data_type) override; 62 port::StatusOr<std::unique_ptr<dnn::RnnSequenceTensorDescriptor>> 67 dnn::DataType data_type) override; [all …]
|
D | cuda_dnn.cc | 205 cudnnDataType_t GetRnnComputeType(dnn::DataType data_type); 207 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() 347 port::StatusOr<perftools::gputools::dnn::VersionInfo> 351 return perftools::gputools::dnn::VersionInfo( in GetVersion() 495 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor() 499 case dnn::DataLayout::kBatchYXDepth: in CudnnTensorDescriptor() 500 case dnn::DataLayout::kBatchDepthYX: { in CudnnTensorDescriptor() [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 41 class MIOpenSupport : public dnn::DnnSupport { 46 port::StatusOr<perftools::gputools::dnn::VersionInfo> GetVersion() override; 48 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor( 50 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, 51 dnn::RnnMode rnn_mode, dnn::DataType data_type, 52 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, 55 port::StatusOr<std::unique_ptr<dnn::RnnSequenceTensorDescriptor>> 58 dnn::DataType data_type) override; 60 port::StatusOr<std::unique_ptr<dnn::RnnStateTensorDescriptor>> 62 dnn::DataType data_type) override; [all …]
|
D | rocm_dnn.cc | 60 using dnn::BatchDescriptor; 61 using dnn::ConvolutionDescriptor; 62 using dnn::FilterDescriptor; 63 using dnn::NormalizeDescriptor; 64 using dnn::PoolingDescriptor; 393 miopenConvFwdAlgorithm_t ToConvForwardAlgo(dnn::AlgorithmDesc algorithm) { in ToConvForwardAlgo() 408 dnn::AlgorithmDesc algorithm) { in ToConvBackwardDataAlgo() 425 dnn::AlgorithmDesc algorithm) { in ToConvBackwardFilterAlgo() 518 port::StatusOr<perftools::gputools::dnn::VersionInfo> 521 return perftools::gputools::dnn::VersionInfo(1, 3, 0); in GetVersion() [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 51 namespace dnn { 344 port::ArraySlice<dnn::BatchDescriptor> inputs); 477 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment); 995 const dnn::BatchDescriptor& x_desc, 996 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 1012 const dnn::BatchDescriptor& x_desc, 1013 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 1043 const dnn::BatchDescriptor& x_desc, 1044 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 1057 const dnn::BatchDescriptor& x_desc, [all …]
|
D | stream.h | 66 namespace dnn { 237 const dnn::BatchDescriptor &x_desc, 238 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 248 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, 249 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 258 const dnn::BatchDescriptor &x_desc, 259 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 270 const dnn::BatchDescriptor &x_desc, 271 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 276 Stream &ThenConvolve(const dnn::BatchDescriptor &input_descriptor, [all …]
|
D | stream.cc | 38 string ToVlogString(const dnn::BatchDescriptor &descriptor) { in ToVlogString() 42 string ToVlogString(const dnn::FilterDescriptor &descriptor) { in ToVlogString() 46 string ToVlogString(const dnn::ConvolutionDescriptor &descriptor) { in ToVlogString() 50 string ToVlogString(const dnn::PoolingDescriptor &descriptor) { in ToVlogString() 54 string ToVlogString(const dnn::NormalizeDescriptor &descriptor) { in ToVlogString() 58 string ToVlogString(dnn::ActivationMode mode) { in ToVlogString() 59 return dnn::ActivationModeString(mode); in ToVlogString() 62 string ToVlogString(const dnn::AlgorithmConfig &algo_config) { in ToVlogString() 66 string ToVlogString(dnn::ElementwiseOperation op) { in ToVlogString() 67 return dnn::ElementwiseOperationString(op); in ToVlogString() [all …]
|
D | stream_executor_pimpl.cc | 325 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveAlgorithms() 326 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms() 337 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetRnnAlgorithms() 338 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms() 347 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardDataAlgorithms() 348 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms() 360 std::vector<dnn::AlgorithmDesc> *out_algorithms) { in GetConvolveBackwardFilterAlgorithms() 361 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms() 380 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> 383 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor() [all …]
|
D | stream_executor_pimpl.h | 386 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); 407 port::StatusOr<std::unique_ptr<dnn::RnnDescriptor>> createRnnDescriptor( 409 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, 410 dnn::RnnMode rnn_mode, dnn::DataType data_type, 411 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed, 416 port::StatusOr<std::unique_ptr<dnn::RnnSequenceTensorDescriptor>> 418 int data_size, dnn::DataType data_type); [all …]
|
D | BUILD | 63 "dnn.h", 104 "dnn.h", 182 "dnn.h", 218 ":dnn", 379 "dnn.h", 434 ":dnn", 461 srcs = ["dnn.proto"], 468 name = "dnn", 469 srcs = ["dnn.cc"], 470 hdrs = ["dnn.h"], [all …]
|
D | dnn.cc | 23 namespace dnn { namespace 53 case dnn::QuantizedActivationMode::k8Bit: in QuantizedActivationModeString() 55 case dnn::QuantizedActivationMode::k16Bit: in QuantizedActivationModeString() 57 case dnn::QuantizedActivationMode::k32Bit: in QuantizedActivationModeString() 146 std::ostream& operator<<(std::ostream& str, dnn::PadAlignment alignment) { in operator <<() 358 port::ArraySlice<dnn::BatchDescriptor> inputs) { in DepthConcatenateOutputDescriptor() 503 : mode_(dnn::PoolingMode::kMaximum), in PoolingDescriptor() 523 mode_ == dnn::PoolingMode::kMaximum ? "kMaximum" : "kAverage"; in ToString() 547 return absl::StrCat(mode_ == dnn::PoolingMode::kMaximum ? "max" : "avg", in ToShortString()
|
D | plugin_registry.cc | 44 blas(kNullPlugin), dnn(kNullPlugin), fft(kNullPlugin), rng(kNullPlugin) { } in DefaultFactories() 128 default_factories_[platform_id].dnn = plugin_id; in SetDefaultFactory() 152 return factories.dnn.find(plugin_id) != factories.dnn.end(); in HasFactory() 243 EMIT_PLUGIN_SPECIALIZATIONS(DnnFactory, dnn, "DNN");
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_pooling_gpu.cc | 35 se::dnn::PoolingMode pooling_mode, in Compute() 71 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() 73 se::dnn::BatchDescriptor input_desc(3); in Compute() 76 .set_layout(se::dnn::DataLayout::kBatchDepthYX); in Compute() 77 se::dnn::BatchDescriptor output_desc(3); in Compute() 80 .set_layout(se::dnn::DataLayout::kBatchDepthYX); in Compute() 82 const auto dim_i = static_cast<se::dnn::DimIndex>(i); in Compute() 119 OpKernelContext* context, se::dnn::PoolingMode pooling_mode, in Compute() 125 CHECK((pooling_mode != se::dnn::PoolingMode::kMaximum) || in Compute() 189 se::dnn::PoolingDescriptor pooling_desc(3); in Compute() [all …]
|
D | pooling_ops_common.cc | 155 se::dnn::PoolingMode pooling_mode, in Compute() 202 se::dnn::DataLayout data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() 206 se::dnn::DataLayout data_layout; in Compute() 209 data_layout = se::dnn::DataLayout::kBatchYXDepth; in Compute() 212 data_layout = se::dnn::DataLayout::kBatchDepthYX; in Compute() 217 data_layout = se::dnn::DataLayout::kBatchYXDepth; in Compute() 228 se::dnn::PoolingDescriptor pooling_desc; in Compute() 238 se::dnn::BatchDescriptor input_desc; in Compute() 245 se::dnn::BatchDescriptor output_desc; in Compute() 286 OpKernelContext* context, se::dnn::PoolingMode pooling_mode, in Compute() [all …]
|
D | conv_ops_fused_impl.h | 449 const se::dnn::ActivationMode activation_mode) 461 std::tuple<ConvParameters::ParameterDataType, se::dnn::ActivationMode>; 468 se::dnn::ActivationMode activation_mode_; 488 se::dnn::AlgorithmConfig>; 507 se::dnn::AlgorithmConfig* algorithm_config) { 514 std::vector<se::dnn::AlgorithmDesc> algorithms; 527 se::dnn::ProfileResult profile_result; 530 launch(se::dnn::AlgorithmConfig(profile_algorithm), &scratch_allocator, 669 se::dnn::ActivationMode dnn_activation_mode; 672 dnn_activation_mode = se::dnn::ActivationMode::kRelu; [all …]
|
D | conv_ops_3d.cc | 39 using stream_executor::dnn::DimIndex; 197 se::dnn::AlgorithmConfig> 351 se::dnn::BatchDescriptor input_desc(3); in launch() 357 .set_layout(se::dnn::DataLayout::kBatchDepthYX); in launch() 358 se::dnn::BatchDescriptor output_desc(3); in launch() 364 .set_layout(se::dnn::DataLayout::kBatchDepthYX); in launch() 365 se::dnn::FilterDescriptor filter_desc(3); in launch() 371 se::dnn::ConvolutionDescriptor conv_desc(3); in launch() 431 using se::dnn::AlgorithmConfig; in launch() 432 using se::dnn::AlgorithmDesc; in launch() [all …]
|
D | gpu_utils.cc | 33 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local 34 se::port::StatusOr<se::dnn::VersionInfo> version_or = dnn->GetVersion(); in GetCudnnVersion() 111 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | stream_executor_util.h | 37 se::dnn::DataLayout input, 38 se::dnn::FilterLayout filter, 39 se::dnn::DataLayout output); 43 std::tuple<se::dnn::DataLayout, se::dnn::FilterLayout, se::dnn::DataLayout>>
|
D | cudnn_batchnorm_thunk.cc | 31 namespace dnn = se::dnn; 33 static std::pair<dnn::BatchDescriptor /*input_desc*/, 34 dnn::BatchDescriptor /*scale_offset_desc*/> 61 dnn::BatchDescriptor input_desc; in MakeDescriptors() 62 input_desc.set_layout(dnn::DataLayout::kBatchDepthYX) in MakeDescriptors() 68 dnn::BatchDescriptor scale_offset_desc; in MakeDescriptors() 69 scale_offset_desc.set_layout(dnn::DataLayout::kBatchDepthYX) in MakeDescriptors() 104 dnn::BatchDescriptor operand_desc; in ExecuteOnStream() 105 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() 166 dnn::BatchDescriptor operand_desc; in ExecuteOnStream() [all …]
|
D | cudnn_conv_runner.cc | 33 using se::dnn::AlgorithmConfig; 34 using se::dnn::BatchDescriptor; 35 using se::dnn::ConvolutionDescriptor; 36 using se::dnn::DataLayout; 37 using se::dnn::DimIndex; 38 using se::dnn::FilterDescriptor; 39 using se::dnn::FilterLayout; 40 using se::dnn::ProfileResult; 56 se::dnn::ActivationMode mode; 72 se::dnn::AlgorithmConfig algorithm; [all …]
|
D | stream_executor_util.cc | 25 using se::dnn::DataLayout; 26 using se::dnn::DataLayoutString; 27 using se::dnn::FilterLayout; 28 using se::dnn::FilterLayoutString;
|
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
D | dnn_test.py | 31 from tensorflow.contrib.learn.python.learn.estimators import dnn 80 dnn._dnn_model_fn(features, labels, model_fn.ModeKeys.TRAIN, params) 114 model_ops = dnn._dnn_model_fn(features, labels, model_fn.ModeKeys.TRAIN, 150 return dnn._dnn_model_fn(features, labels, model_fn.ModeKeys.TRAIN, params) 169 estimator=dnn.DNNClassifier( 181 estimator_test_utils.assert_estimator_contract(self, dnn.DNNEstimator) 206 dnn_estimator = dnn.DNNEstimator( 221 estimator=dnn.DNNClassifier( 237 estimator_test_utils.assert_estimator_contract(self, dnn.DNNClassifier) 244 classifier = dnn.DNNClassifier( [all …]
|
D | nonlinear_test.py | 25 from tensorflow.contrib.learn.python.learn.estimators import dnn 41 classifier = dnn.DNNClassifier( 67 regressor = dnn.DNNRegressor( 97 classifier = dnn.DNNClassifier( 109 classifier = dnn.DNNClassifier( 121 classifier = dnn.DNNClassifier(
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 260 namespace dnn = se::dnn; 275 if (auto* dnn = stream_executor->AsDnn()) { in GetCudnnVersion() local 276 se::port::StatusOr<se::dnn::VersionInfo> version_or = dnn->GetVersion(); in GetCudnnVersion() 344 se::dnn::AlgorithmConfig* algo) { in BestCudnnConvAlgorithm() 392 dnn::AlgorithmConfig> 574 constexpr auto data_layout = is_int8x4 ? dnn::DataLayout::kBatchDepthYX4 in launch() 575 : dnn::DataLayout::kBatchDepthYX; in launch() 576 constexpr auto filter_layout = is_int8x4 ? dnn::FilterLayout::kOutputInputYX4 in launch() 577 : dnn::FilterLayout::kOutputInputYX; in launch() 581 dnn::BatchDescriptor conv_input_desc; in launch() [all …]
|
/external/tensorflow/tensorflow/python/estimator/canned/ |
D | dnn.py | 26 from tensorflow_estimator.python.estimator.canned import dnn 30 dnn.__all__ = [s for s in dir(dnn) if not s.startswith('__')] 32 from tensorflow_estimator.python.estimator.canned.dnn import *
|