Home
last modified time | relevance | path

Searched refs:is_cost_viable (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dconvolution_group_converter.h30 ConvolutionGroupConverter(std::function<bool(HloInstruction*)> is_cost_viable,
33 : is_cost_viable_(is_cost_viable), in is_cost_viable_() argument
Dconvolution_group_converter.cc57 std::function<bool(HloInstruction*)> is_cost_viable,
69 std::function<bool(HloInstruction*)> is_cost_viable, in ConvolutionVisitor() argument
75 is_cost_viable_(is_cost_viable) {} in ConvolutionVisitor()
95 std::function<bool(HloInstruction*)> is_cost_viable, in Run() argument
97 ConvolutionVisitor visitor(computation, is_cost_viable, in Run()