Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dlaunch_dimensions.cc57 int64 ThreadsPerBlockRowVectorized(const Shape& shape, in ThreadsPerBlockRowVectorized() function
105 ThreadsPerBlockRowVectorized(shape, gpu_device_info, dim_config); in CalculateLaunchDimensions()
Dlaunch_dimensions.h102 int64 ThreadsPerBlockRowVectorized(const Shape& shape,
Dir_emitter_unnested.cc1717 ThreadsPerBlockRowVectorized(element_shape, in EmitLoopFusion()