Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h353 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
366 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Dcuda_dnn.cc2985 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoPrepareForConvolution()
3041 const dnn::BatchDescriptor& input_descriptor, DeviceMemoryBase input_data, in DoConvolve()
4005 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBiasImpl()
4026 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4037 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4048 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
4198 CudnnTensorDescriptor input_descriptor(dimensions, CUDNN_DATA_FLOAT); in DoBiasAdd() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.h327 const dnn::BatchDescriptor &input_descriptor, in ConvolveWithAlgorithm()
442 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardDataWithAlgorithm()
469 const dnn::BatchDescriptor &input_descriptor, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1305 bool DoConvolve(Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolve()
1413 const dnn::BatchDescriptor& input_descriptor, in DoConvolveBackwardData()
1455 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardFilter()
1495 const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
1503 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
1511 Stream* stream, const BatchDescriptor& input_descriptor, in DoConvolveBackwardBias()
Dstream.cc603 const dnn::BatchDescriptor &input_descriptor, in ThenConvolve()
622 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
649 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveQuantized()
702 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardBiasImpl()
720 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardBias()
729 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardBias()
738 const dnn::BatchDescriptor &input_descriptor, in ThenConvolveBackwardBias()
Dstream_executor_pimpl.cc279 const dnn::BatchDescriptor &input_descriptor, DeviceMemoryBase input_data, in GetMIOpenConvolveAlgorithms()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h377 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
390 Stream* stream, const dnn::BatchDescriptor& input_descriptor, in DoConvolveQuantized()
Drocm_dnn.cc435 miopenTensorDescriptor_t input_descriptor) { in FindOrCreate()
1031 const miopenTensorDescriptor_t input_descriptor) in ScopedFusionPlanBase()
1051 miopenStatus_t Execute(miopenTensorDescriptor_t input_descriptor, in Execute()
1231 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1311 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1347 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationInference()
1417 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1447 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationForward()
1517 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in GetFusionOpHashValue()
1548 miopenHandle_t miopen_handle, miopenTensorDescriptor_t input_descriptor, in ScopedFusionPlanBatchNormActivationBackward()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_runner.h70 se::dnn::BatchDescriptor input_descriptor; member
Dgpu_conv_runner.cc374 BatchDescriptor& input_descriptor = config.input_descriptor; in GetGpuConvConfig() local