Searched defs:output_descriptor (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 359 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized() 372 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized() 385 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
|
D | cuda_dnn.cc | 2987 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/ |
D | rocm_dnn.h | 383 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized() 396 const dnn::BatchDescriptor& output_descriptor, in DoConvolveQuantized() 409 const dnn::BatchDescriptor& output_descriptor, in DoSeparableConvolve()
|
D | rocm_dnn.cc | 1053 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/ |
D | stream.h | 332 const dnn::BatchDescriptor &output_descriptor, in ConvolveWithAlgorithm() 439 const dnn::BatchDescriptor &output_descriptor, in ConvolveBackwardDataWithAlgorithm() 471 const dnn::BatchDescriptor &output_descriptor, in ConvolveBackwardFilterWithAlgorithm()
|
D | dnn.h | 1152 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()
|
D | stream.cc | 455 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()
|
D | stream_executor_pimpl.cc | 281 DeviceMemoryBase filter_data, const dnn::BatchDescriptor &output_descriptor, in GetMIOpenConvolveAlgorithms()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_runner.h | 72 se::dnn::BatchDescriptor output_descriptor; member
|
D | gpu_conv_runner.cc | 416 BatchDescriptor& output_descriptor = config.output_descriptor; in GetGpuConvConfig() local
|