Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.h366 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
379 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
392 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Dcuda_dnn.cc670 CudnnFilterDescriptor(const dnn::FilterDescriptor& filter_descriptor, in CudnnFilterDescriptor()
3507 const dnn::FilterDescriptor& filter_descriptor, in GetCudnnOperationGraph()
3633 const dnn::FilterDescriptor& filter_descriptor, in GetCudnnFusedOperationGraph()
3959 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
4260 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
4715 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners()
4820 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc()
5068 const dnn::FilterDescriptor& filter_descriptor, in FusedConvolveRunnerFromDesc()
5143 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners()
5758 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h370 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
383 const dnn::FilterDescriptor& filter_descriptor, in DoConvolveQuantized()
396 const dnn::FilterDescriptor& filter_descriptor, int depth_multiplier, in DoSeparableConvolve()
Drocm_dnn.cc714 ScopedFilterDescriptor(const FilterDescriptor& filter_descriptor, in ScopedFilterDescriptor()
1286 miopenTensorDescriptor_t filter_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1366 miopenTensorDescriptor_t filter_descriptor, in GetFusionOpHashValue()
2906 const dnn::FilterDescriptor& filter_descriptor, in DoPrepareForConvolution()
2958 FilterDescriptor filter_descriptor, in RocmConvRunner()
3115 const dnn::FilterDescriptor& filter_descriptor, in DoConvolve()
3150 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners()
3190 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc()
3215 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
3237 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dlazy_op_runner.h124 const FilterDescriptor& filter_descriptor; member
148 const FilterDescriptor& filter_descriptor; member
Dstream.h339 const dnn::FilterDescriptor &filter_descriptor, in ConvolveWithAlgorithm()
370 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveWithAlgorithm()
397 const dnn::FilterDescriptor &filter_descriptor, in ConvolveRunnerFromDesc()
416 const dnn::FilterDescriptor &filter_descriptor, in FusedConvolveRunnerFromDesc()
Ddnn.cc136 const dnn::FilterDescriptor& filter_descriptor, in ConvolveRunnerFromDesc()
148 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners()
164 const dnn::FilterDescriptor& filter_descriptor, in FusedConvolveRunnerFromDesc()
Dstream_executor_pimpl.cc297 const dnn::FilterDescriptor& filter_descriptor, in GetConvolveRunners()
320 const dnn::FilterDescriptor& filter_descriptor, in GetFusedConvolveRunners()
340 const dnn::FilterDescriptor& filter_descriptor, in GetMIOpenConvolveAlgorithms()
Ddnn.h1305 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolve()
1324 const FilterDescriptor& filter_descriptor, in PrepareForConvolution()
2424 const dnn::FilterDescriptor& filter_descriptor, in DoFusedConvolutionBiasActivation()
2622 const FilterDescriptor& filter_descriptor, DeviceMemoryBase filter_data, in DoPrepareForConvolution()
Dstream.cc440 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolve()
460 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
487 const dnn::FilterDescriptor &filter_descriptor, in ThenConvolveQuantized()
514 const dnn::FilterDescriptor &filter_descriptor, int depth_multiplier, in ThenSeparableConvolve()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h62 se::dnn::FilterDescriptor filter_descriptor; member
Dgpu_conv_runner.cc407 FilterDescriptor& filter_descriptor = config.filter_descriptor; in GetGpuConvConfig() local