Home
last modified time | relevance | path

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

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