Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h334 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
347 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Dcuda_dnn.cc2627 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
2685 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
3472 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBiasImpl()
3493 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
3504 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
3515 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
3665 CudnnTensorDescriptor input_descriptor(dimensions, CUDNN_DATA_FLOAT); in DoBiasAdd() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h325 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
338 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Drocm_dnn.cc320 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate()
914 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase()
934 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute()
1118 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1198 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1234 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1304 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1334 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1404 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1435 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_conv_runner.cc181 BatchDescriptor input_descriptor(effective_num_dimensions); in RunCudnnConvImpl() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc594 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
636 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
678 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveWithAlgorithm()
720 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve()
735 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
765 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
827 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
872 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
917 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardDataWithAlgorithm()
957 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardFilterWithAlgorithm()
[all …]