Searched defs:ComputationInstructionOrdering (Results 1 – 1 of 1) sorted by relevance
27 struct ComputationInstructionOrdering { struct28 explicit ComputationInstructionOrdering(const HloComputation& computation) { in ComputationInstructionOrdering() function43 int32_t NodeIdForInstruction(const HloInstruction& instr) { in NodeIdForInstruction()57 bool InsertEdge(const HloInstruction& source, const HloInstruction& dest) { in InsertEdge()63 absl::flat_hash_map<int32_t, int32_t> node_id_to_graph_id;65 tensorflow::GraphCycles graph_cycles;