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.h33 : is_cost_viable_(is_cost_viable), in is_cost_viable_() function
47 std::function<bool(HloInstruction*)> is_cost_viable_; variable
Dconvolution_group_converter.cc75 is_cost_viable_(is_cost_viable) {} in ConvolutionVisitor()
90 std::function<bool(HloInstruction*)> is_cost_viable_; member in xla::__anon9c1c05530111::ConvolutionVisitor
227 if (!is_cost_viable_(convolution) || filter_expansion_) { in HandleBatchGroupCount()
546 if (ConvolutionVisitor::Run(comp, is_cost_viable_, in Run()