Searched defs:BatchDescriptor (Results 1 – 3 of 3) sorted by relevance
119 Stream* /*stream*/, const dnn::BatchDescriptor& /*input_descriptor*/, in GetConvolveRunners()175 Stream* /*stream*/, const dnn::BatchDescriptor& /*input_descriptor*/, in GetMIOpenConvolveAlgorithms()461 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor469 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
67 class BatchDescriptor; variable
6492 const dnn::BatchDescriptor&, in DoElementwiseOperate()