Home
last modified time | relevance | path

Searched defs:output_descriptor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h340 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
353 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
366 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc2629 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2687 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
2931 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolveImpl()
3352 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3376 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3401 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3425 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h331 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
344 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
357 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc936 miopenTensorDescriptor_t output_descriptor, in Execute()
2595 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2775 BatchDescriptor* output_descriptor, DeviceMemoryBase backward_output_data, in MaybeTransformLayout()
2813 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3154 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3171 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3189 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3206 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
4030 const dnn::BatchDescriptor& output_descriptor, DeviceMemory<T>* output_data, in DoFusedConvolutionBiasActivationImpl()
4118 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc218 BatchDescriptor output_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc449 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<double> *output, in ThenFusedConvolveWithAlgorithm()
487 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, in ThenFusedConvolveWithAlgorithm()
526 const dnn::BatchDescriptor &output_descriptor, in ThenFusedConvolveWithAlgorithm()
564 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<int8> *output, in ThenFusedConvolveWithAlgorithm()
599 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<double> *output, in ThenConvolveWithAlgorithm()
641 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, in ThenConvolveWithAlgorithm()
683 const dnn::BatchDescriptor &output_descriptor, in ThenConvolveWithAlgorithm()
725 const dnn::BatchDescriptor &output_descriptor, in ThenConvolve()
741 const dnn::BatchDescriptor &output_descriptor, in ThenConvolveQuantized()
771 const dnn::BatchDescriptor &output_descriptor, in ThenConvolveQuantized()
[all …]