Home
last modified time | relevance | path

Searched refs:conv_input_descriptor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h265 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,
Drocm_dnn.cc3146 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/
Dcuda_dnn.h274 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,
Dcuda_dnn.cc2923 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/
Ddnn.h1116 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,
Dstream.cc441 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 …]
Dstream.h338 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,