Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h281 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
295 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
309 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
325 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
339 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
617 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
Dcuda_dnn.cc3178 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolveImpl() argument
3199 conv_input_descriptor, in DoFusedConvolveImpl()
3200 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()
3203 GetCudnnDataType<OutputType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()
3206 GetCudnnDataType<ElementType>(conv_input_descriptor.layout())); in DoFusedConvolveImpl()
3757 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
3770 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3779 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
3792 stream, conv_input_descriptor, conv_input_data, conv_input_scale, in DoFusedConvolve()
3801 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h319 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
333 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
347 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
363 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
586 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor,
783 const dnn::BatchDescriptor& conv_input_descriptor,
Drocm_dnn.cc3676 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
3692 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
3708 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
3725 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
4619 const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl() argument
4631 conv_input_descriptor, static_cast<miopenDataType_t>(miopen_type)}; in DoFusedConvolutionBiasActivationImpl()
4642 ScopedFilterDescriptor filter{filter_descriptor, conv_input_descriptor, in DoFusedConvolutionBiasActivationImpl()
4707 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation() argument
4718 stream, miopenFloat, conv_input_descriptor, conv_input_data, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1144 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
1162 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
1181 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
1202 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolve() argument
2488 Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, in DoFusedConvolutionBiasActivation() argument
Dstream.cc447 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() argument
459 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
468 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
478 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() argument
490 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
499 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
509 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() argument
522 VLOG_CALL(PARAM(conv_input_descriptor), PARAM(conv_input_data), in FusedConvolveWithAlgorithm()
531 this, conv_input_descriptor, conv_input_data, conv_input_scale, in FusedConvolveWithAlgorithm()
541 const dnn::BatchDescriptor &conv_input_descriptor, in FusedConvolveWithAlgorithm() argument
[all …]
Dstream.h355 const dnn::BatchDescriptor &conv_input_descriptor,
369 const dnn::BatchDescriptor &conv_input_descriptor,
383 const dnn::BatchDescriptor &conv_input_descriptor,
398 const dnn::BatchDescriptor &conv_input_descriptor,
412 const dnn::BatchDescriptor &conv_input_descriptor,