Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dlaunch_dimensions.cc63 int64_t threads_per_block_row_vectorized = in ThreadsPerBlockRowVectorized() local
72 threads_per_block_row_vectorized <= in ThreadsPerBlockRowVectorized()
74 return threads_per_block_row_vectorized; in ThreadsPerBlockRowVectorized()
101 int64_t threads_per_block_row_vectorized = in CalculateLaunchDimensions() local
111 threads_per_block_row_vectorized > 0 in CalculateLaunchDimensions()
112 ? threads_per_block_row_vectorized in CalculateLaunchDimensions()
123 threads_per_block_row_vectorized > 0 && in CalculateLaunchDimensions()
124 threads_per_block_row_vectorized < 128 && num_elements > 128 in CalculateLaunchDimensions()
126 threads_per_block_row_vectorized) in CalculateLaunchDimensions()