Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dpartition_assignment.cc42 int64 ThreadsPerBlockLimit(const se::DeviceDescription& device_desc) { in ThreadsPerBlockLimit() function
84 int64 threads_per_block = ThreadsPerBlockLimit(device_desc); in CalculateLaunchDimensions()
Dpartition_assignment.h61 int64 ThreadsPerBlockLimit(const se::DeviceDescription& device_desc);
Dir_emitter_unnested.cc3687 ThreadsPerBlockLimit(ir_emitter_context_->device_description()); in ComputeMappingSchemeAndReductionKind()