Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h359 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
372 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Dcuda_dnn.cc2895 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2954 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
4014 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBiasImpl()
4035 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4046 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4057 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4207 CudnnTensorDescriptor input_descriptor(dimensions, CUDNN_DATA_FLOAT); in DoBiasAdd() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h344 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
357 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Drocm_dnn.cc359 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate()
962 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase()
982 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute()
1162 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1242 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1278 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1348 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1378 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1448 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1479 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h62 se::dnn::BatchDescriptor input_descriptor; member
Dgpu_conv_runner.cc378 BatchDescriptor& input_descriptor = params.input_descriptor; in GetGpuConvParams() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc652 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
694 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
736 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
778 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
820 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
862 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve()
877 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
907 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
969 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
1014 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
[all …]
Dstream_executor_pimpl.cc295 const dnn::BatchDescriptor &input_descriptor, in GetMIOpenConvolveAlgorithms()