Searched refs:bias_nd (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2952 CudnnTensorDescriptor bias_nd(bias_descriptor, GetCudnnDataType<BiasType>()); in DoFusedConvolveImpl() local
2998 << "\nbias_nd.handle() = " << bias_nd.handle() in DoFusedConvolveImpl()
3021 /*biasDesc=*/bias_nd.handle(), /*bias=*/biases.opaque(), in DoFusedConvolveImpl()
3478 CudnnTensorDescriptor bias_nd(bias_descriptor, cudnn_type); in DoConvolveBackwardBiasImpl() local
3488 bias_nd.handle(), backward_bias_data->opaque())); in DoConvolveBackwardBiasImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3237 ScopedTensorDescriptor bias_nd{bias_descriptor, in DoConvolveBackwardBiasImpl() local
3247 bias_nd.handle(), backward_bias_data->opaque()); in DoConvolveBackwardBiasImpl()
4037 ScopedTensorDescriptor bias_nd{bias_descriptor, in DoFusedConvolutionBiasActivationImpl() local
4053 conv.handle(), bias_nd.handle(), activation_desc}; in DoFusedConvolutionBiasActivationImpl()
|