| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 213 const dnn::BatchDescriptor& x_desc,
214 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
226 const dnn::BatchDescriptor& x_desc,
227 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
238 const dnn::BatchDescriptor& x_desc,
239 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
247 const dnn::BatchDescriptor& x_desc,
248 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
255 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
258 const dnn::BatchDescriptor& output_descriptor,
[all …]
|
| D | rocm_dnn.cc | 60 using dnn::BatchDescriptor;
527 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor()
595 const BatchDescriptor& batch_descriptor, in ScopedFilterDescriptor()
2593 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2595 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2775 BatchDescriptor* output_descriptor, DeviceMemoryBase backward_output_data, in MaybeTransformLayout()
2784 BatchDescriptor transformed_output_descriptor; in MaybeTransformLayout()
2811 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
2813 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
2868 BatchDescriptor output_back_descriptor; in DoConvolve()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 222 const dnn::BatchDescriptor& x_desc,
223 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
235 const dnn::BatchDescriptor& x_desc,
236 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
247 const dnn::BatchDescriptor& x_desc,
248 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
256 const dnn::BatchDescriptor& x_desc,
257 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
264 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
267 const dnn::BatchDescriptor& output_descriptor,
[all …]
|
| D | cuda_dnn.cc | 495 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor()
2600 bool ShouldIncludeWinogradNonfusedAlgo(const dnn::BatchDescriptor&, in ShouldIncludeWinogradNonfusedAlgo() argument
2601 const dnn::BatchDescriptor&) { in ShouldIncludeWinogradNonfusedAlgo()
2606 const dnn::BatchDescriptor& input_desc, in ShouldIncludeWinogradNonfusedAlgo()
2607 const dnn::BatchDescriptor& output_desc) { in ShouldIncludeWinogradNonfusedAlgo()
2627 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2629 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2685 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
2687 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
2923 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 238 class BatchDescriptor {
242 BatchDescriptor();
243 explicit BatchDescriptor(int ndims);
246 void CloneFrom(const BatchDescriptor& other);
279 BatchDescriptor& set_count(int64 value) {
283 BatchDescriptor& set_feature_map_count(int64 value) {
287 BatchDescriptor& set_height(int64 value) {
291 BatchDescriptor& set_width(int64 value) {
295 BatchDescriptor& set_spatial_dim(DimIndex dim, int64 value) {
299 BatchDescriptor& set_value_max(float value) {
[all …]
|
| D | dnn.cc | 238 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
246 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
248 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const { in full_dims()
257 std::vector<int64> BatchDescriptor::full_strides( in full_strides()
276 void BatchDescriptor::CloneFrom(const BatchDescriptor& other) { in CloneFrom()
283 string BatchDescriptor::ToString() const { in ToString()
295 string BatchDescriptor::ToShortString() const { in ToShortString()
332 int64 BatchDescriptor::NodesPerFeatureMap() const { in NodesPerFeatureMap()
340 int64 BatchDescriptor::NodesAcrossFeatureMaps() const { in NodesAcrossFeatureMaps()
344 int64 BatchDescriptor::ElementCount() const { in ElementCount()
[all …]
|
| D | stream.h | 67 class BatchDescriptor; variable
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()
344 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
345 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
370 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
371 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
394 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
395 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
421 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
422 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
441 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm()
[all …]
|
| D | dnn.proto | 54 // BatchDescriptor::value_max().
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 33 static std::pair<dnn::BatchDescriptor /*input_desc*/,
34 dnn::BatchDescriptor /*scale_offset_desc*/>
61 dnn::BatchDescriptor input_desc; in MakeDescriptors()
68 dnn::BatchDescriptor scale_offset_desc; in MakeDescriptors()
104 dnn::BatchDescriptor operand_desc; in ExecuteOnStream()
105 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream()
166 dnn::BatchDescriptor operand_desc; in ExecuteOnStream()
167 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream()
250 dnn::BatchDescriptor operand_desc; in ExecuteOnStream()
251 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream()
|
| D | cudnn_conv_runner.cc | 34 using se::dnn::BatchDescriptor;
181 BatchDescriptor input_descriptor(effective_num_dimensions); in RunCudnnConvImpl()
218 BatchDescriptor output_descriptor(effective_num_dimensions); in RunCudnnConvImpl()
273 BatchDescriptor bias_desc; in RunCudnnConvImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 73 se::dnn::BatchDescriptor input_desc(3); in Compute()
77 se::dnn::BatchDescriptor output_desc(3); in Compute()
192 se::dnn::BatchDescriptor orig_output_desc(3); in Compute()
197 se::dnn::BatchDescriptor orig_input_desc(3); in Compute()
|
| D | pooling_ops_common.cc | 238 se::dnn::BatchDescriptor input_desc; in Compute()
245 se::dnn::BatchDescriptor output_desc; in Compute()
386 se::dnn::BatchDescriptor orig_output_desc; in Compute()
393 se::dnn::BatchDescriptor orig_input_desc; in Compute()
|
| D | fused_batch_norm_op.cc | 303 se::dnn::BatchDescriptor x_desc; in operator ()()
310 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
457 se::dnn::BatchDescriptor x_desc; in operator ()()
464 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
|
| D | conv_ops_fused_impl.h | 678 se::dnn::BatchDescriptor input_desc;
689 se::dnn::BatchDescriptor bias_desc;
703 se::dnn::BatchDescriptor output_desc;
|
| D | conv_ops_3d.cc | 351 se::dnn::BatchDescriptor input_desc(3); in launch()
358 se::dnn::BatchDescriptor output_desc(3); in launch()
|
| D | lrn_op.cc | 194 se::dnn::BatchDescriptor dimensions_desc; in launch()
411 se::dnn::BatchDescriptor dimensions_desc; in launch()
|
| D | conv_grad_ops_3d.cc | 1254 se::dnn::BatchDescriptor input_desc(3); in Compute()
1261 se::dnn::BatchDescriptor output_desc(3); in Compute()
1650 se::dnn::BatchDescriptor input_desc(3); in Compute()
1660 se::dnn::BatchDescriptor output_desc(3); in Compute()
|
| D | conv_ops.cc | 763 se::dnn::BatchDescriptor input_desc; in operator ()()
769 se::dnn::BatchDescriptor output_desc; in operator ()()
|
| D | conv_grad_filter_ops.cc | 716 se::dnn::BatchDescriptor input_desc; in operator ()()
722 se::dnn::BatchDescriptor output_desc; in operator ()()
|
| D | conv_grad_input_ops.cc | 832 se::dnn::BatchDescriptor input_desc; in operator ()()
838 se::dnn::BatchDescriptor output_desc; in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 581 dnn::BatchDescriptor conv_input_desc; in launch()
593 dnn::BatchDescriptor side_input_desc; in launch()
599 dnn::BatchDescriptor bias_desc; in launch()
605 dnn::BatchDescriptor output_desc; in launch()
|