Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc90 params.config.filter_descriptor, filter_buf, params.config.conv_desc, in RunGpuConvForward()
130 params.config.conv_result_scale, params.config.filter_descriptor, in RunGpuConvForwardActivation()
167 params.config.filter_descriptor, filter_buf, in RunGpuConvInternalImpl()
181 params.config.filter_descriptor, &filter_buf, scratch_allocator, in RunGpuConvInternalImpl()
387 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig() local
388 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()
389 filter_descriptor.set_layout(filter_dl) in GetGpuConvConfig()
395 filter_descriptor.set_spatial_dim( in GetGpuConvConfig()
432 filter_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig()
Dgpu_conv_runner.h71 se::dnn::FilterDescriptor filter_descriptor; member
Dgpu_conv_algorithm_picker.cc144 params.config.filter_descriptor, params.filter_buf, in GetMIOpenAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h237 const dnn::FilterDescriptor& filter_descriptor,
310 const dnn::FilterDescriptor& filter_descriptor,
321 const dnn::FilterDescriptor& filter_descriptor,
335 const dnn::FilterDescriptor& filter_descriptor,
349 const dnn::FilterDescriptor& filter_descriptor,
365 const dnn::FilterDescriptor& filter_descriptor,
379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
392 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
405 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
574 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Drocm_dnn.cc710 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor() argument
722 std::vector<int> dims(2 + filter_descriptor.ndims()); in ScopedFilterDescriptor()
723 dims[0] = filter_descriptor.output_feature_map_count(); in ScopedFilterDescriptor()
724 dims[1] = filter_descriptor.input_feature_map_count(); in ScopedFilterDescriptor()
725 const auto& spatial_dims = filter_descriptor.input_filter_dims(); in ScopedFilterDescriptor()
1232 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1239 filter_descriptor, conv_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1247 fusion_plan_, &conv_op, conv_descriptor, filter_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
1312 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue() argument
1324 tensorflow::Hash64Combine(hash_value, GetHashValue(filter_descriptor)); in GetFusionOpHashValue()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h272 const dnn::FilterDescriptor& filter_descriptor,
283 const dnn::FilterDescriptor& filter_descriptor,
297 const dnn::FilterDescriptor& filter_descriptor,
311 const dnn::FilterDescriptor& filter_descriptor,
327 const dnn::FilterDescriptor& filter_descriptor,
341 const dnn::FilterDescriptor& filter_descriptor,
355 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
368 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
381 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
558 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dcuda_dnn.cc567 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor() argument
575 switch (filter_descriptor.layout()) { in CudnnFilterDescriptor()
587 << FilterLayoutString(filter_descriptor.layout()); in CudnnFilterDescriptor()
591 std::vector<int> dims(2 + filter_descriptor.ndims()); in CudnnFilterDescriptor()
592 dims[0] = filter_descriptor.output_feature_map_count(); in CudnnFilterDescriptor()
593 dims[1] = filter_descriptor.input_feature_map_count(); in CudnnFilterDescriptor()
595 filter_descriptor.input_filter_dims(); in CudnnFilterDescriptor()
2986 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution() argument
2997 filter_descriptor, in DoPrepareForConvolution()
2998 ToCudnnDataType(element_type, filter_descriptor.layout())); in DoPrepareForConvolution()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1146 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve() argument
1164 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve() argument
1183 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve() argument
1204 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve() argument
1245 const FilterDescriptor& filter_descriptor, in PrepareForConvolution() argument
1255 input_data, filter_descriptor, filter_data, output_descriptor, in PrepareForConvolution()
1297 DeviceMemoryBase input_data, const FilterDescriptor& filter_descriptor,
1307 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve() argument
1318 input_data, filter_descriptor, filter_data, in DoConvolve()
1333 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dstream.h299 const dnn::FilterDescriptor &filter_descriptor,
308 const dnn::FilterDescriptor &filter_descriptor,
318 const dnn::FilterDescriptor &filter_descriptor,
329 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm() argument
341 filter_descriptor, filter_data, output_descriptor, *output, in ConvolveWithAlgorithm()
347 input_data, filter_descriptor, filter_data, output_descriptor, in ConvolveWithAlgorithm()
357 const dnn::FilterDescriptor &filter_descriptor,
371 const dnn::FilterDescriptor &filter_descriptor,
385 const dnn::FilterDescriptor &filter_descriptor,
400 const dnn::FilterDescriptor &filter_descriptor,
[all …]
Dstream.cc449 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm() argument
460 PARAM(conv_input_scale), PARAM(filter_descriptor), in FusedConvolveWithAlgorithm()
469 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
480 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm() argument
491 PARAM(conv_input_scale), PARAM(filter_descriptor), in FusedConvolveWithAlgorithm()
500 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
511 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm() argument
523 PARAM(conv_input_scale), PARAM(filter_descriptor), in FusedConvolveWithAlgorithm()
532 filter_descriptor, filter_data, convolution_descriptor, side_input_data, in FusedConvolveWithAlgorithm()
543 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream_executor_pimpl.cc280 const dnn::FilterDescriptor &filter_descriptor, in GetMIOpenConvolveAlgorithms() argument
292 filter_descriptor, filter_data, output_descriptor, output_data, in GetMIOpenConvolveAlgorithms()
Dstream_executor_pimpl.h373 const dnn::FilterDescriptor &filter_descriptor,