Home
last modified time | relevance | path

Searched refs:DnnBatchDescriptors (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc65 struct DnnBatchDescriptors { struct
70 DnnBatchDescriptors MakeBatchNormDescriptors(const Shape& shape, in MakeBatchNormDescriptors()
97 DnnBatchDescriptors batch_descs; in MakeBatchNormDescriptors()
123 DnnBatchDescriptors batch_descs = in AssignCommonParams()