Home
last modified time | relevance | path

Searched refs:dnn (Results 1 – 25 of 78) sorted by relevance

1234

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h43 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 …]
Dcuda_dnn.cc205 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/
Drocm_dnn.h41 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 …]
Drocm_dnn.cc60 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/
Ddnn.h51 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 …]
Dstream.h66 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 …]
Dstream.cc38 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 …]
Dstream_executor_pimpl.cc325 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 …]
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);
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 …]
DBUILD63 "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 …]
Ddnn.cc23 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()
Dplugin_registry.cc44 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/
Dcudnn_pooling_gpu.cc35 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 …]
Dpooling_ops_common.cc155 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 …]
Dconv_ops_fused_impl.h449 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 …]
Dconv_ops_3d.cc39 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 …]
Dgpu_utils.cc33 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/
Dstream_executor_util.h37 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>>
Dcudnn_batchnorm_thunk.cc31 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 …]
Dcudnn_conv_runner.cc33 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 …]
Dstream_executor_util.cc25 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/
Ddnn_test.py31 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 …]
Dnonlinear_test.py25 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/
Dfused_conv2d_bias_activation_op.cc260 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/
Ddnn.py26 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 *

1234