| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 50 bool IsSame(const dnn::BatchDescriptor& input_dimensions,
51 const dnn::BatchDescriptor& output_dimensions,
61 bool find(const void* p, const dnn::BatchDescriptor& input_dimensions,
62 const dnn::BatchDescriptor& output_dimensions,
65 void insert(const void* p, const dnn::BatchDescriptor& input_dimensions,
66 const dnn::BatchDescriptor& output_dimensions,
236 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
239 const dnn::BatchDescriptor& output_descriptor,
261 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc,
262 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
[all …]
|
| D | rocm_dnn.cc | 64 using dnn::BatchDescriptor;
647 ScopedTensorDescriptor(const BatchDescriptor& batch_descriptor, in ScopedTensorDescriptor()
711 const BatchDescriptor& batch_descriptor, in ScopedFilterDescriptor()
2845 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2847 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2904 BatchDescriptor* output_descriptor, DeviceMemoryBase backward_output_data, in MaybeTransformLayout()
2913 BatchDescriptor transformed_output_descriptor; in MaybeTransformLayout()
2941 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
2943 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3003 BatchDescriptor output_back_descriptor; in DoConvolve()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 223 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc,
224 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
237 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc,
238 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
250 const dnn::BatchDescriptor& x_desc,
251 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
261 const dnn::BatchDescriptor& x_desc,
262 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
271 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data,
274 const dnn::BatchDescriptor& output_descriptor,
[all …]
|
| D | cuda_dnn.cc | 517 CudnnTensorDescriptor(const dnn::BatchDescriptor& batch_descriptor, in CudnnTensorDescriptor()
2985 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2987 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
3041 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
3043 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3178 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl()
3184 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolveImpl()
3186 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolveImpl()
3452 const DeviceMemory<float>& side_input, const dnn::BatchDescriptor& x_desc, in DoBatchNormalizationForward()
3453 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 230 class BatchDescriptor {
234 BatchDescriptor();
235 explicit BatchDescriptor(int ndims);
238 void CloneFrom(const BatchDescriptor& other);
271 BatchDescriptor& set_count(int64 value) { in set_count()
275 BatchDescriptor& set_feature_map_count(int64 value) { in set_feature_map_count()
279 BatchDescriptor& set_height(int64 value) { in set_height()
283 BatchDescriptor& set_width(int64 value) { in set_width()
287 BatchDescriptor& set_spatial_dim(DimIndex dim, int64 value) { in set_spatial_dim()
291 BatchDescriptor& set_value_max(float value) { in set_value_max()
[all …]
|
| D | dnn.cc | 52 Stream* /*stream*/, const dnn::BatchDescriptor& /*input_descriptor*/, in GetMIOpenConvolveAlgorithms() argument
56 const dnn::BatchDescriptor& /*output_descriptor*/, in GetMIOpenConvolveAlgorithms()
269 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
277 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
279 std::vector<int64> BatchDescriptor::full_dims(const DataLayout& layout) const { in full_dims()
288 std::vector<int64> BatchDescriptor::full_strides( in full_strides()
307 void BatchDescriptor::CloneFrom(const BatchDescriptor& other) { in CloneFrom()
314 std::string BatchDescriptor::ToString() const { in ToString()
326 std::string BatchDescriptor::ToShortString() const { in ToShortString()
363 int64 BatchDescriptor::NodesPerFeatureMap() const { in NodesPerFeatureMap()
[all …]
|
| D | stream.h | 68 class BatchDescriptor; variable
251 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc,
252 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
264 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc,
265 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
276 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc,
277 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
290 const dnn::BatchDescriptor &x_desc,
291 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
297 Stream &ThenConvolve(const dnn::BatchDescriptor &input_descriptor,
[all …]
|
| D | stream.cc | 37 std::string ToVlogString(const dnn::BatchDescriptor &descriptor) { in ToVlogString()
347 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
348 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
373 const DeviceMemory<float> &inv_var, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
374 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
398 const DeviceMemory<float> &side_input, const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationForward()
399 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
425 const dnn::BatchDescriptor &x_desc, in ThenBatchNormalizationBackward()
426 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
447 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm()
[all …]
|
| D | dnn.proto | 58 // BatchDescriptor::value_max().
|
| D | stream_executor_pimpl.h | 372 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data,
375 const dnn::BatchDescriptor &output_descriptor,
|
| D | stream_executor_pimpl.cc | 279 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms()
281 DeviceMemoryBase filter_data, const dnn::BatchDescriptor &output_descriptor, in GetMIOpenConvolveAlgorithms()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_utils.h | 222 const se::dnn::BatchDescriptor& input_desc,
224 const se::dnn::BatchDescriptor& output_desc,
234 const se::dnn::BatchDescriptor& input_desc,
236 const se::dnn::BatchDescriptor& output_desc,
|
| D | gpu_utils.cc | 131 const se::dnn::BatchDescriptor& input_desc, in LogConvAutotuneResults()
133 const se::dnn::BatchDescriptor& output_desc, in LogConvAutotuneResults()
174 const se::dnn::BatchDescriptor& input_desc, in LogFusedConvForwardAutotuneResults()
176 const se::dnn::BatchDescriptor& output_desc, in LogFusedConvForwardAutotuneResults()
|
| D | cudnn_pooling_gpu.cc | 73 se::dnn::BatchDescriptor input_desc(3); in Compute()
77 se::dnn::BatchDescriptor output_desc(3); in Compute()
207 se::dnn::BatchDescriptor orig_output_desc(3); in Compute()
212 se::dnn::BatchDescriptor orig_input_desc(3); in Compute()
|
| D | pooling_ops_common.cc | 362 se::dnn::BatchDescriptor input_desc; in Compute()
369 se::dnn::BatchDescriptor output_desc; in Compute()
677 se::dnn::BatchDescriptor orig_output_desc; in Compute()
684 se::dnn::BatchDescriptor orig_input_desc; in Compute()
|
| D | lrn_op.cc | 203 se::dnn::BatchDescriptor dimensions_desc; in launch()
256 perftools::gputools::dnn::BatchDescriptor dimensions_desc; in launch()
495 se::dnn::BatchDescriptor dimensions_desc; in launch()
570 perftools::gputools::dnn::BatchDescriptor dimensions_desc; in launch()
|
| D | conv_ops_fused_impl.h | 572 se::dnn::BatchDescriptor input_desc;
583 se::dnn::BatchDescriptor bias_desc;
597 se::dnn::BatchDescriptor output_desc;
|
| D | conv_grad_input_ops.cc | 249 se::dnn::BatchDescriptor input_desc; in operator ()()
255 se::dnn::BatchDescriptor output_desc; in operator ()()
|
| D | conv_ops_3d.cc | 397 se::dnn::BatchDescriptor input_desc(3); in launch()
404 se::dnn::BatchDescriptor output_desc(3); in launch()
|
| D | fused_batch_norm_op.cc | 890 se::dnn::BatchDescriptor x_desc; in operator ()()
897 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
1081 se::dnn::BatchDescriptor x_desc; in operator ()()
1088 se::dnn::BatchDescriptor scale_offset_desc; in operator ()()
|
| D | conv_grad_ops_3d.cc | 1298 se::dnn::BatchDescriptor input_desc(3); in Compute()
1305 se::dnn::BatchDescriptor output_desc(3); in Compute()
1792 se::dnn::BatchDescriptor input_desc(3); in Compute()
1802 se::dnn::BatchDescriptor output_desc(3); in Compute()
|
| D | conv_grad_filter_ops.cc | 854 se::dnn::BatchDescriptor input_desc; in operator ()()
860 se::dnn::BatchDescriptor output_desc; in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.h | 70 se::dnn::BatchDescriptor input_descriptor;
72 se::dnn::BatchDescriptor output_descriptor;
|
| D | cudnn_batchnorm_runner.cc | 33 se::dnn::BatchDescriptor operand_desc;
34 se::dnn::BatchDescriptor scale_offset_desc;
66 se::dnn::BatchDescriptor input_desc;
67 se::dnn::BatchDescriptor scale_offset_desc;
|
| D | gpu_conv_runner.cc | 35 using se::dnn::BatchDescriptor;
103 BatchDescriptor bias_desc; in RunGpuConvForwardActivation()
374 BatchDescriptor& input_descriptor = config.input_descriptor; in GetGpuConvConfig()
375 input_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()
416 BatchDescriptor& output_descriptor = config.output_descriptor; in GetGpuConvConfig()
417 output_descriptor = BatchDescriptor(effective_num_dimensions); in GetGpuConvConfig()
|