Home
last modified time | relevance | path

Searched defs:bias_descriptor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dlazy_op_runner.h149 const BatchDescriptor& bias_descriptor; member
Dstream.h374 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveWithAlgorithm()
417 const dnn::BatchDescriptor &bias_descriptor, in FusedConvolveRunnerFromDesc()
Ddnn.cc149 const dnn::BatchDescriptor& bias_descriptor, in GetFusedConvolveRunners()
165 const dnn::BatchDescriptor& bias_descriptor, in FusedConvolveRunnerFromDesc()
Dstream_executor_pimpl.cc321 const dnn::BatchDescriptor& bias_descriptor, in GetFusedConvolveRunners()
Ddnn.h1309 const dnn::BatchDescriptor& bias_descriptor, DeviceMemoryBase biases, in DoFusedConvolve()
2427 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1288 miopenTensorDescriptor_t bias_descriptor, in ScopedFusionPlanConvolutionBiasActivation()
1368 miopenTensorDescriptor_t bias_descriptor, in GetFusionOpHashValue()
3804 const dnn::BatchDescriptor& bias_descriptor, DeviceMemoryBase biases, in DoFusedConvolve()
3976 ScopedTensorDescriptor bias_descriptor{bias_dimensions, miopenFloat}; in DoBiasAdd() local
4563 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolutionBiasActivationImpl()
4651 const dnn::BatchDescriptor& bias_descriptor, in DoFusedConvolutionBiasActivation()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc3634 dnn::BatchDescriptor bias_descriptor, in GetCudnnFusedOperationGraph()
5069 const dnn::BatchDescriptor& bias_descriptor, in FusedConvolveRunnerFromDesc()
5144 const dnn::BatchDescriptor& bias_descriptor, in GetFusedConvolveRunners()
5762 const dnn::BatchDescriptor& bias_descriptor, DeviceMemoryBase biases, in DoFusedConvolve()
6105 CudnnTensorDescriptor bias_descriptor(bias_dimensions, CUDNN_DATA_FLOAT); in DoBiasAdd() local