/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 265 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 279 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 293 const dnn::BatchDescriptor& conv_input_descriptor, 311 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 534 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 708 const dnn::BatchDescriptor& conv_input_descriptor,
|
D | rocm_dnn.cc | 3146 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 3163 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 3180 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 3198 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 4023 const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl() argument 4035 conv_input_descriptor, static_cast<miopenDataType_t>(miopen_type)}; in DoFusedConvolutionBiasActivationImpl() 4046 ScopedFilterDescriptor filter{filter_descriptor, conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl() 4111 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation() argument 4122 stream, miopenFloat, conv_input_descriptor, conv_input_data, in DoFusedConvolutionBiasActivation()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 274 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 288 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 302 const dnn::BatchDescriptor& conv_input_descriptor, 320 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 583 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
|
D | cuda_dnn.cc | 2923 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl() argument 2944 conv_input_descriptor, in DoFusedConvolveImpl() 2945 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl() 2948 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl() 2951 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl() 3005 !ShouldIncludeWinogradNonfusedAlgo(conv_input_descriptor, in DoFusedConvolveImpl() 3344 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 3358 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve() 3368 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument 3382 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve() [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1116 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 1133 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 1151 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 1171 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, 2353 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
|
D | stream.cc | 441 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() argument 453 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm() 463 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm() 479 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() argument 491 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm() 501 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm() 517 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() argument 530 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in ThenFusedConvolveWithAlgorithm() 540 this, conv_input_descriptor, conv_input_data, conv_input_scale, in ThenFusedConvolveWithAlgorithm() 556 const dnn::BatchDescriptor &conv_input_descriptor, in ThenFusedConvolveWithAlgorithm() argument [all …]
|
D | stream.h | 338 const dnn::BatchDescriptor &conv_input_descriptor, 352 const dnn::BatchDescriptor &conv_input_descriptor, 366 const dnn::BatchDescriptor &conv_input_descriptor, 381 const dnn::BatchDescriptor &conv_input_descriptor,
|