Home
last modified time | relevance | path

Searched defs:input_descriptor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h336 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
349 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Dcuda_dnn.cc3143 const BatchDescriptor& input_descriptor, in DoConvolveBackwardDataImpl()
3353 const BatchDescriptor& input_descriptor, in DoConvolveBackwardData()
3371 const BatchDescriptor& input_descriptor, in DoConvolveBackwardData()
3385 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardFilterImpl()
3596 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardFilter()
3614 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardFilter()
3633 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBiasImpl()
3666 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
3675 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
3684 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_convolution_runner.cc118 BatchDescriptor input_descriptor(effective_num_dimensions); in RunCudnnConvolution() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc516 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithScratch()
543 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithScratch()
685 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
716 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
765 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve()
779 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
809 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
871 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithScratch()
899 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
932 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
[all …]