Searched defs:should_expand (Results 1 – 4 of 4) sorted by relevance
51 auto should_expand = [](HloInstruction* conv) { return true; }; in TEST_F() local84 auto should_expand = [](HloInstruction* conv) { return true; }; in TEST_F() local113 auto should_expand = [](HloInstruction* conv) { return true; }; in TEST_F() local142 auto should_expand = [](HloInstruction* conv) { return true; }; in TEST_F() local
70 std::function<bool(HloInstruction*)> should_expand, in ConvolutionVisitor()97 std::function<bool(HloInstruction*)> should_expand, in Run()
36 : should_expand_(should_expand), in should_expand_() argument
32 bool should_expand(const IntArrayRef &from_size, const IntArrayRef &to_size) { in should_expand() function