Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Transforms/
H A Dgpu_fusion_rewrite.cc133 static int64_t getElementsPerThread(TensorType type) { in getElementsPerThread() function
234 int64_t unrollFactor = getElementsPerThread(resultType); in matchAndRewrite()