Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconvolution_thunk.cc33 const HloCustomCallInstruction* cudnn_call, in ConvolutionThunk() argument
37 : Thunk(Kind::kConvolution, cudnn_call), in ConvolutionThunk()
38 cudnn_call_(cudnn_call), in ConvolutionThunk()
Dconvolution_thunk.h46 ConvolutionThunk(const HloCustomCallInstruction* cudnn_call,