Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dlaunch_dimensions.cc28 const LaunchDimensions& launch_dims) { in operator <<() argument
29 LaunchDimensions::Dim3D block_counts = launch_dims.block_counts(); in operator <<()
30 LaunchDimensions::Dim3D thread_counts = launch_dims.thread_counts_per_block(); in operator <<()
Dkernel_thunk.cc65 void KernelThunk::SetLaunchDimensions(const LaunchDimensions& launch_dims) { in SetLaunchDimensions() argument
67 launch_dimensions_ = launch_dims; in SetLaunchDimensions()
Dkernel_thunk.h58 void SetLaunchDimensions(const LaunchDimensions& launch_dims);
Dlaunch_dimensions.h66 const LaunchDimensions& launch_dims);
Dir_emitter_unnested.cc158 void UpdateLaunchDimensions(const LaunchDimensions& launch_dims, Thunk* thunk, in UpdateLaunchDimensions() argument
162 kernel_thunk->SetLaunchDimensions(launch_dims); in UpdateLaunchDimensions()
173 launch_dims.thread_counts_per_block().x); in UpdateLaunchDimensions()