Searched refs:conv_input_nd (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2943 CudnnTensorDescriptor conv_input_nd( in DoFusedConvolveImpl() local
2964 stream, cudnn, algorithm_config, conv_input_nd, filter, conv, in DoFusedConvolveImpl()
2987 << "\nconv_input_nd.handle() = " << conv_input_nd.handle() in DoFusedConvolveImpl()
3015 /*srcDesc=*/conv_input_nd.handle(), /*srcData=*/conv_input_data.opaque(), in DoFusedConvolveImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4034 ScopedTensorDescriptor conv_input_nd{ in DoFusedConvolutionBiasActivationImpl() local
4052 miopen.handle(), conv_input_nd.handle(), filter.handle(), in DoFusedConvolutionBiasActivationImpl()
4083 fusion_plan.Execute(conv_input_nd.handle(), conv_input_data.opaque(), in DoFusedConvolutionBiasActivationImpl()
|