Home
last modified time | relevance | path

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

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