Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc75 auto physical_dim_size = [&](int64 physical_dim) { in MakeBatchNormDescriptors() local
90 batch_size *= physical_dim_size(physical_dim); in MakeBatchNormDescriptors()
94 y_size *= physical_dim_size(physical_dim); in MakeBatchNormDescriptors()