| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 383 FilterDescriptor::FilterDescriptor(int ndims) { in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
388 FilterDescriptor::FilterDescriptor() : FilterDescriptor(/*ndims=*/2) {} in FilterDescriptor() function in stream_executor::dnn::FilterDescriptor
390 FilterDescriptor::~FilterDescriptor() {} in ~FilterDescriptor()
392 void FilterDescriptor::CloneFrom(const FilterDescriptor& other) { in CloneFrom()
396 string FilterDescriptor::ToString() const { in ToString()
410 string FilterDescriptor::ToShortString() const { in ToShortString()
439 int64 FilterDescriptor::ComputeWeightCount() const { in ComputeWeightCount()
447 TensorDescriptorProto FilterDescriptor::ToProto(DataType data_type) const { in ToProto()
|
| D | dnn.h | 389 class FilterDescriptor {
394 FilterDescriptor();
395 explicit FilterDescriptor(int ndims);
396 ~FilterDescriptor();
399 FilterDescriptor& set_output_feature_map_count(int64 value) {
403 FilterDescriptor& set_input_feature_map_count(int64 value) {
407 FilterDescriptor& set_input_filter_height(int64 value) {
411 FilterDescriptor& set_input_filter_width(int64 value) {
415 FilterDescriptor& set_layout(FilterLayout layout) {
419 FilterDescriptor& set_spatial_dim(DimIndex dim, int64 value) {
[all …]
|
| D | stream.h | 68 class FilterDescriptor; variable
278 const dnn::FilterDescriptor &filter_descriptor,
287 const dnn::FilterDescriptor &filter_descriptor,
297 const dnn::FilterDescriptor &filter_descriptor,
307 const dnn::FilterDescriptor &filter_descriptor,
318 const dnn::FilterDescriptor &filter_descriptor,
329 const dnn::FilterDescriptor &filter_descriptor,
340 const dnn::FilterDescriptor &filter_descriptor,
354 const dnn::FilterDescriptor &filter_descriptor,
368 const dnn::FilterDescriptor &filter_descriptor,
[all …]
|
| D | stream.cc | 42 string ToVlogString(const dnn::FilterDescriptor &descriptor) { in ToVlogString()
443 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
481 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
519 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
558 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm()
596 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
638 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
680 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveWithAlgorithm()
722 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
737 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 265 const dnn::FilterDescriptor& filter_descriptor,
276 const dnn::FilterDescriptor& filter_descriptor,
290 const dnn::FilterDescriptor& filter_descriptor,
305 const dnn::FilterDescriptor& filter_descriptor,
322 const dnn::FilterDescriptor& filter_descriptor,
336 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
349 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
362 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
539 const dnn::FilterDescriptor& filter_descriptor,
586 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| D | cuda_dnn.cc | 419 using FilterDescriptor = typedef
447 FilterDescriptor CreateFilterDescriptor() { in CreateFilterDescriptor()
450 return FilterDescriptor(result); in CreateFilterDescriptor()
545 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
583 FilterDescriptor handle_; // Owned.
968 CudnnRnnParamsDescriptor(FilterDescriptor handle, int64 params_size_in_bytes, in CudnnRnnParamsDescriptor()
993 FilterDescriptor handle_;
1177 FilterDescriptor filter_desc = CreateFilterDescriptor(); in Create()
1201 FilterDescriptor region_desc_handle = CreateFilterDescriptor(); in Create()
2628 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 256 const dnn::FilterDescriptor& filter_descriptor,
267 const dnn::FilterDescriptor& filter_descriptor,
281 const dnn::FilterDescriptor& filter_descriptor,
296 const dnn::FilterDescriptor& filter_descriptor,
313 const dnn::FilterDescriptor& filter_descriptor,
327 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
340 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
353 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
522 const dnn::FilterDescriptor& filter_descriptor,
536 const dnn::FilterDescriptor& filter_descriptor,
[all …]
|
| D | rocm_dnn.cc | 62 using dnn::FilterDescriptor;
594 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
2594 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2812 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3148 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3165 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3182 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3200 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
3988 const FilterDescriptor& filter_descriptor, in DeriveOutputBatchDescriptor()
4025 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolutionBiasActivationImpl()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 38 using se::dnn::FilterDescriptor;
193 FilterDescriptor filter_descriptor(effective_num_dimensions); in RunCudnnConvImpl()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 365 se::dnn::FilterDescriptor filter_desc(3); in launch()
|
| D | conv_grad_ops_3d.cc | 1268 se::dnn::FilterDescriptor filter_desc(3); in Compute()
1667 se::dnn::FilterDescriptor filter_desc(3); in Compute()
|
| D | conv_ops_fused_impl.h | 684 se::dnn::FilterDescriptor filter_desc;
|
| D | conv_ops.cc | 775 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| D | conv_grad_filter_ops.cc | 728 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| D | conv_grad_input_ops.cc | 844 se::dnn::FilterDescriptor filter_desc; in operator ()()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 587 dnn::FilterDescriptor filter_desc; in launch()
|