Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_algorithm_picker.h31 class GemmAlgorithmPicker : public HloModulePass {
33 GemmAlgorithmPicker(se::StreamExecutor* stream_exec, in GemmAlgorithmPicker() function
H A Dgemm_algorithm_picker.cc380 StatusOr<bool> GemmAlgorithmPicker::Run( in Run()
H A Dnvptx_compiler.cc155 post_pipeline.AddPass<GemmAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()