Searched defs:BatchDescriptor (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ | ||
D | stream.h | 67 class BatchDescriptor; variable |
D | dnn.cc | 238 BatchDescriptor::BatchDescriptor(int ndims) in BatchDescriptor() function in stream_executor::dnn::BatchDescriptor |
/external/tensorflow/tensorflow/stream_executor/cuda/ | ||
D | cuda_dnn.cc | 2600 bool ShouldIncludeWinogradNonfusedAlgo(const dnn::BatchDescriptor&, in ShouldIncludeWinogradNonfusedAlgo() |