Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dlazy_op_runner.h124 const FilterDescriptor& filter_descriptor; member
134 config.input_descriptor, config.filter_descriptor, in RunnerFromAlgorithmDesc()
148 const FilterDescriptor& filter_descriptor; member
162 config.filter_descriptor, config.bias_descriptor, in RunnerFromAlgorithmDesc()
Ddnn.h1305 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve() argument
1324 const FilterDescriptor& filter_descriptor, in PrepareForConvolution() argument
1334 input_data, filter_descriptor, filter_data, output_descriptor, in PrepareForConvolution()
1376 DeviceMemoryBase input_data, const FilterDescriptor& filter_descriptor,
1393 const dnn::FilterDescriptor& filter_descriptor,
1406 const dnn::FilterDescriptor& filter_descriptor,
1416 const dnn::FilterDescriptor& filter_descriptor,
1430 const dnn::FilterDescriptor& filter_descriptor,
1439 const dnn::FilterDescriptor& filter_descriptor,
1458 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dstream.h309 const dnn::FilterDescriptor &filter_descriptor,
318 const dnn::FilterDescriptor &filter_descriptor,
328 const dnn::FilterDescriptor &filter_descriptor,
339 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm() argument
351 kind, this, input_descriptor, input_data, filter_descriptor, in ConvolveWithAlgorithm()
357 input_descriptor, input_data, filter_descriptor, in ConvolveWithAlgorithm()
370 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm() argument
385 conv_input_data, conv_input_scale, filter_descriptor, filter_data, in FusedConvolveWithAlgorithm()
397 const dnn::FilterDescriptor &filter_descriptor, in ConvolveRunnerFromDesc() argument
406 filter_descriptor, output_descriptor, convolution_descriptor); in ConvolveRunnerFromDesc()
[all …]
Dstream_executor_pimpl.cc297 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners() argument
309 input_descriptor, input_data, filter_descriptor, filter_data, in GetConvolveRunners()
320 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners() argument
333 input_descriptor, filter_descriptor, bias_descriptor, output_descriptor, in GetFusedConvolveRunners()
340 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms() argument
352 filter_descriptor, filter_data, output_descriptor, output_data, in GetMIOpenConvolveAlgorithms()
Dstream_executor_pimpl.h376 const dnn::FilterDescriptor& filter_descriptor,
390 const dnn::FilterDescriptor& filter_descriptor,
402 const dnn::FilterDescriptor& filter_descriptor,
Dstream.cc440 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve() argument
448 filter_descriptor, filter_data, output_descriptor, *output, in ThenConvolve()
460 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized() argument
467 PARAM(filter_descriptor), PARAM(filter_coefficients), in ThenConvolveQuantized()
473 this, input_descriptor, input_data, filter_descriptor, in ThenConvolveQuantized()
487 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized() argument
494 PARAM(filter_descriptor), PARAM(filter_coefficients), in ThenConvolveQuantized()
500 this, input_descriptor, input_data, filter_descriptor, in ThenConvolveQuantized()
514 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier, in ThenSeparableConvolve() argument
521 PARAM(batch_descriptor), PARAM(input_data), PARAM(filter_descriptor), in ThenSeparableConvolve()
[all …]
Ddnn.cc136 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc() argument
148 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners() argument
164 const dnn::FilterDescriptor& filter_descriptor, in FusedConvolveRunnerFromDesc() argument
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h244 const dnn::FilterDescriptor& filter_descriptor,
257 const dnn::FilterDescriptor& filter_descriptor,
264 const dnn::FilterDescriptor& filter_descriptor,
343 const dnn::FilterDescriptor& filter_descriptor,
356 const dnn::FilterDescriptor& filter_descriptor,
370 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
383 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
396 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
511 const dnn::FilterDescriptor& filter_descriptor,
525 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Drocm_dnn.cc714 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor() argument
756 switch (filter_descriptor.layout()) { in ScopedFilterDescriptor()
759 const int nd = filter_descriptor.ndims() + 2; in ScopedFilterDescriptor()
763 filter_descriptor.full_strides(dnn::FilterLayout::kOutputInputYX); in ScopedFilterDescriptor()
765 filter_descriptor.full_dims(dnn::FilterLayout::kOutputInputYX); in ScopedFilterDescriptor()
779 << filter_descriptor.ToString() in ScopedFilterDescriptor()
785 << FilterLayoutString(filter_descriptor.layout()); in ScopedFilterDescriptor()
1286 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation() argument
1293 miopen_handle, input_descriptor, filter_descriptor, conv_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1301 fusion_plan_, &conv_op, conv_descriptor, filter_descriptor); in ScopedFusionPlanConvolutionBiasActivation()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h224 const dnn::FilterDescriptor& filter_descriptor,
237 const dnn::FilterDescriptor& filter_descriptor,
247 const dnn::FilterDescriptor& filter_descriptor,
262 const dnn::FilterDescriptor& filter_descriptor,
339 const dnn::FilterDescriptor& filter_descriptor,
352 const dnn::FilterDescriptor& filter_descriptor,
366 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized() argument
392 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve() argument
505 const dnn::FilterDescriptor& filter_descriptor,
[all …]
Dcuda_dnn.cc670 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor() argument
678 switch (filter_descriptor.layout()) { in CudnnFilterDescriptor()
688 filter_descriptor.layout() == dnn::FilterLayout::kOutputInputYX4 in CudnnFilterDescriptor()
697 << FilterLayoutString(filter_descriptor.layout()); in CudnnFilterDescriptor()
701 std::vector<int> dims(2 + filter_descriptor.ndims()); in CudnnFilterDescriptor()
702 dims[0] = filter_descriptor.output_feature_map_count(); in CudnnFilterDescriptor()
703 dims[1] = filter_descriptor.input_feature_map_count(); in CudnnFilterDescriptor()
705 filter_descriptor.input_filter_dims(); in CudnnFilterDescriptor()
3507 const dnn::FilterDescriptor& filter_descriptor, in GetCudnnOperationGraph() argument
3547 GetTensorVectorSizeAndDim(filter_descriptor, input_type); in GetCudnnOperationGraph()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.cc105 params.config->filter_descriptor, in RunGpuConvUnfused()
165 params.config->filter_descriptor, in RunGpuConvForwardActivation()
407 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig() local
408 filter_descriptor = FilterDescriptor(effective_num_dimensions); in GetGpuConvConfig()
409 filter_descriptor.set_layout(filter_dl) in GetGpuConvConfig()
416 filter_descriptor.set_spatial_dim( in GetGpuConvConfig()
454 filter_descriptor.set_spatial_dim(static_cast<DimIndex>(dim), 1); in GetGpuConvConfig()
Dgpu_conv_runner.h62 se::dnn::FilterDescriptor filter_descriptor; member
Dgpu_conv_algorithm_picker.cc143 config.filter_descriptor, GetBiasDescriptor(config), in GetAlgorithms()
166 config.filter_descriptor, in GetAlgorithms()
206 params.config->filter_descriptor, params.filter_buf, in GetMIOpenAlgorithms()