Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h359 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
372 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
385 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
Dcuda_dnn.cc2987 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
3043 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3186 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolveImpl()
3765 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3787 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3810 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3832 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3863 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h383 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
396 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized()
409 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
Drocm_dnn.cc1053 miopenTensorDescriptor_t output_descriptor, in Execute()
2847 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoPrepareForConvolution()
2904 BatchDescriptor* output_descriptor, DeviceMemoryBase backward_output_data, in MaybeTransformLayout()
2943 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
3101 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithms()
3123 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode()
3332 DeviceMemoryBase filter_data, const dnn::BatchDescriptor& output_descriptor, in GetMIOpenConvolveAlgorithmsFindMode()
3684 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3700 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
3717 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Dstream.h332 const dnn::BatchDescriptor &output_descriptor, in ConvolveWithAlgorithm()
439 const dnn::BatchDescriptor &output_descriptor, in ConvolveBackwardDataWithAlgorithm()
471 const dnn::BatchDescriptor &output_descriptor, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1152 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
1170 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
1190 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
1210 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolve()
1247 const BatchDescriptor& output_descriptor, in PrepareForConvolution()
1310 const dnn::BatchDescriptor& output_descriptor, in DoConvolve()
1410 const dnn::BatchDescriptor& output_descriptor, in DoConvolveBackwardData()
1457 const BatchDescriptor& output_descriptor, in DoConvolveBackwardFilter()
2495 const dnn::BatchDescriptor& output_descriptor, in DoFusedConvolutionBiasActivation()
2683 const BatchDescriptor& output_descriptor, DeviceMemoryBase output_data, in DoPrepareForConvolution()
Dstream.cc455 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<double> *output, in FusedConvolveWithAlgorithm()
486 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, in FusedConvolveWithAlgorithm()
518 const dnn::BatchDescriptor &output_descriptor, in FusedConvolveWithAlgorithm()
549 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<int8> *output, in FusedConvolveWithAlgorithm()
580 const dnn::BatchDescriptor &output_descriptor, DeviceMemory<float> *output, in FusedConvolveWithAlgorithm()
608 const dnn::BatchDescriptor &output_descriptor, in ThenConvolve()
628 const dnn::BatchDescriptor &output_descriptor, in ThenConvolveQuantized()
655 const dnn::BatchDescriptor &output_descriptor, in ThenConvolveQuantized()
682 const dnn::BatchDescriptor &output_descriptor, in ThenSeparableConvolve()
Dstream_executor_pimpl.cc281 DeviceMemoryBase filter_data, const dnn::BatchDescriptor &output_descriptor, in GetMIOpenConvolveAlgorithms()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h72 se::dnn::BatchDescriptor output_descriptor; member
Dgpu_conv_runner.cc416 BatchDescriptor& output_descriptor = config.output_descriptor; in GetGpuConvConfig() local