Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_query.cc107 bool ContainsInstrWithOpcode(const HloComputation* comp, in ContainsInstrWithOpcode() function
114 if (ContainsInstrWithOpcode(subcomp, opcodes)) { in ContainsInstrWithOpcode()
Dhlo_query.h49 bool ContainsInstrWithOpcode(const HloComputation* comp,
Dwhile_loop_simplifier.cc34 using hlo_query::ContainsInstrWithOpcode;
1000 if (ContainsInstrWithOpcode(while_op->while_body(), in Run()
1003 ContainsInstrWithOpcode(while_op->while_condition(), in Run()
1026 if (ContainsInstrWithOpcode(while_op->while_body(), {HloOpcode::kDomain}) || in Run()
1027 ContainsInstrWithOpcode(while_op->while_condition(), in Run()