Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_precision_insertion.cc236 std::vector<bool> opcode_filter(HloOpcodeCount(), false); in make_filter_function() local
238 opcode_filter[opcode] = true; in make_filter_function()
241 return [opcode_filter](const HloInstruction* instruction) { in make_filter_function()
242 return opcode_filter[static_cast<unsigned int>(instruction->opcode())]; in make_filter_function()
250 return [opcode_filter, in make_filter_function()
252 if (!opcode_filter[static_cast<unsigned int>(instruction->opcode())]) { in make_filter_function()