Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dlaunch_dimensions.cc57 int64_t ThreadsPerBlockRowVectorized(const Shape& shape, in ThreadsPerBlockRowVectorized() function
102 ThreadsPerBlockRowVectorized(shape, gpu_device_info, dim_config); in CalculateLaunchDimensions()
H A Dlaunch_dimensions.h111 int64_t ThreadsPerBlockRowVectorized(const Shape& shape,
H A Dir_emitter_unnested.cc1724 ThreadsPerBlockRowVectorized(element_shape, in EmitLoopFusion()