Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_fusible.cc303 int num_variadic = in SharedMemoryUsageNoCache() local
307 return 32 * primitive_size * num_variadic; in SharedMemoryUsageNoCache()
311 return 2 * 32 * 33 * primitive_size * num_variadic; in SharedMemoryUsageNoCache()