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_thunk.cc39 auto physical_dim_size = [&](int64 physical_dim) { in MakeDescriptors() local
54 batch_size *= physical_dim_size(physical_dim); in MakeDescriptors()
58 y_size *= physical_dim_size(physical_dim); in MakeDescriptors()