Home
last modified time | relevance | path

Searched defs:launch_dims (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dlaunch_dimensions.cc29 const LaunchDimensions& launch_dims) { in operator <<()
Dir_emitter_unnested.cc174 void AnnotateThunkLaunchDimensions(const LaunchDimensions& launch_dims, in AnnotateThunkLaunchDimensions()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dkernel_ops_pattern.cc433 const auto& launch_dims = kernel_thunk->launch_dimensions(); in Rewrite() local