Home
last modified time | relevance | path

Searched refs:filter_descriptor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc193 FilterDescriptor filter_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
194 filter_descriptor.set_layout(filter_dl) in RunCudnnConvImpl()
200 filter_descriptor.set_spatial_dim( in RunCudnnConvImpl()
233 filter_descriptor.set_spatial_dim(static_cast<DimIndex>(0), 1); in RunCudnnConvImpl()
246 input_descriptor, input_buf, filter_descriptor, filter_buf, in RunCudnnConvImpl()
257 filter_descriptor, filter_buf, output_descriptor, output_buf, in RunCudnnConvImpl()
269 convolution_descriptor, filter_descriptor, &filter_buf, in RunCudnnConvImpl()
300 filter_descriptor, filter_buf, convolution_descriptor, side_input, in RunCudnnConvImpl()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h265 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() argument
349 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
362 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
539 const dnn::FilterDescriptor& filter_descriptor,
586 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dcuda_dnn.cc545 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor() argument
553 switch (filter_descriptor.layout()) { in CudnnFilterDescriptor()
565 << FilterLayoutString(filter_descriptor.layout()); in CudnnFilterDescriptor()
569 std::vector<int> dims(2 + filter_descriptor.ndims()); in CudnnFilterDescriptor()
570 dims[0] = filter_descriptor.output_feature_map_count(); in CudnnFilterDescriptor()
571 dims[1] = filter_descriptor.input_feature_map_count(); in CudnnFilterDescriptor()
573 filter_descriptor.input_filter_dims(); in CudnnFilterDescriptor()
2628 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution() argument
2639 filter_descriptor, in DoPrepareForConvolution()
2640 ToCudnnDataType(element_type, filter_descriptor.layout())); in DoPrepareForConvolution()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1118 const dnn::FilterDescriptor& filter_descriptor,
1135 const dnn::FilterDescriptor& filter_descriptor,
1153 const dnn::FilterDescriptor& filter_descriptor,
1173 const dnn::FilterDescriptor& filter_descriptor,
1191 const FilterDescriptor& filter_descriptor,
1201 input_data, filter_descriptor, filter_data, output_descriptor,
1243 const FilterDescriptor& filter_descriptor, DeviceMemoryBase filter_data,
1252 const dnn::FilterDescriptor& filter_descriptor,
1262 stream, input_descriptor, input_data, filter_descriptor,
1286 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dstream.cc443 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm() argument
454 PARAM(conv_input_scale), PARAM(filter_descriptor), in ThenFusedConvolveWithAlgorithm()
464 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
481 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm() argument
492 PARAM(conv_input_scale), PARAM(filter_descriptor), in ThenFusedConvolveWithAlgorithm()
502 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
519 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm() argument
531 PARAM(conv_input_scale), PARAM(filter_descriptor), in ThenFusedConvolveWithAlgorithm()
541 filter_descriptor, filter_data, convolution_descriptor, in ThenFusedConvolveWithAlgorithm()
558 const dnn::FilterDescriptor &filter_descriptor, in ThenFusedConvolveWithAlgorithm() argument
[all …]
Dstream.h278 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,
383 const dnn::FilterDescriptor &filter_descriptor,
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h256 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() argument
340 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
353 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
522 const dnn::FilterDescriptor& filter_descriptor,
536 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Drocm_dnn.cc594 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor() argument
611 std::vector<int> dims(2 + filter_descriptor.ndims()); in ScopedFilterDescriptor()
612 dims[0] = filter_descriptor.output_feature_map_count(); in ScopedFilterDescriptor()
613 dims[1] = filter_descriptor.input_feature_map_count(); in ScopedFilterDescriptor()
614 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor()
1119 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1126 filter_descriptor, conv_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1134 fusion_plan_, &conv_op, conv_descriptor, filter_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1199 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue() argument
1211 tensorflow::Hash64Combine(hash_value, GetHashValue(filter_descriptor)); in GetFusionOpHashValue()
[all …]