Home
last modified time | relevance | path

Searched defs:BatchDescriptor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc52 Stream* /*stream*/, const dnn::BatchDescriptor& /*input_descriptor*/, in GetMIOpenConvolveAlgorithms()
269 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
277 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
Dstream.h68 class BatchDescriptor; variable
Ddnn.h1221 Stream* /*stream*/, const dnn::BatchDescriptor& /*conv_input_descriptor*/, in DoFusedConvolve()