Searched defs:output_units (Results 1 – 2 of 2) sorted by relevance
439 def output_units(unit): function
428 const int output_units = output_shape.Dims(thread_dim); in HowManyConvThreads() local