Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream.h67 class BatchDescriptor; variable
Ddnn.cc238 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
246 BatchDescriptor::BatchDescriptor() : BatchDescriptor(/*ndims=*/2) {} in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2600 bool ShouldIncludeWinogradNonfusedAlgo(const dnn::BatchDescriptor&, in ShouldIncludeWinogradNonfusedAlgo()