Home
last modified time | relevance | path

Searched refs:output_nd_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc4157 /*yDesc=*/output_nd_.handle(), /*y=*/output_data.opaque())); in operator ()()
4167 /*dyDesc=*/output_nd_.handle(), in operator ()()
4185 /*diffDesc=*/output_nd_.handle(), in operator ()()
4232 output_nd_(std::move(output_nd)), in CudnnLegacyConvRunner()
4251 CudnnTensorDescriptor output_nd_; member in stream_executor::gpu::CudnnLegacyConvRunner
4959 << "\noutput_nd.handle() = " << output_nd_.handle() in operator ()()
4964 << "\noutput_nd.handle() = " << output_nd_.handle() in operator ()()
4985 /*zDesc=*/output_nd_.handle(), /*z=*/side_input_data_ptr, in operator ()()
4988 /*yDesc=*/output_nd_.handle(), /*y=*/output_data.opaque()); in operator ()()
5034 output_nd_(std::move(output_nd)), in CudnnLegacyFusedConvRunner()
[all …]