input_nd (definition) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched defs:input_nd (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A D	cuda_dnn.cc	2764 const CudnnHandle& cudnn, const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionForwardAlgo() 2803 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardDataAlgo() 2846 const CudnnTensorDescriptor& input_nd, in GetCudnnConvolutionBackwardFilterAlgo() 2888 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionForwardWorkspace() 2937 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardDataWorkspace() 2987 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in AllocateCudnnConvolutionBackwardFilterWorkspace() 3055 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in GetCudnnConvolutionForwardAlgorithm() 3118 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardDataAlgorithm() 3179 const CudnnTensorDescriptor& input_nd, const CudnnFilterDescriptor& filter, in GetCudnnConvolutionBackwardFilterAlgorithm() 3966 CudnnTensorDescriptor input_nd( in DoPrepareForConvolution() local [all …]
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A D	rocm_dnn.cc	`3245 ScopedTensorDescriptor input_nd{input_descriptor, in GetMIOpenConvolveAlgorithmsImmediateMode() local 3454 ScopedTensorDescriptor input_nd{input_descriptor, in GetMIOpenConvolveAlgorithmsFindMode() local 4528 ScopedTensorDescriptor input_nd{batch_descriptor, miopenFloat}; in DeriveOutputBatchDescriptor() local`