Searched refs:pass_id (Results 1 – 7 of 7) sorted by relevance
101 StatusOr<int64_t> pass_id = module.metadata()->current_pass_id(); in SetInstructionMetadata() local102 if (!pass_id.ok()) { in SetInstructionMetadata()103 LOG(FATAL) << pass_id.status(); in SetInstructionMetadata()108 instruction->set_creation_pass_id(*pass_id); in SetInstructionMetadata()111 instruction->set_logical_creation_pass_id(*pass_id); in SetInstructionMetadata()
80 static_cast<int64_t>(added_pass_metadata->pass_id()) + 1); in set_prepartitioning_metadata()
67 return pass_metadata->pass_id(); in current_pass_id()
681 // various other stats like duration. `pass_id` may also be used to identify a685 // For a given module, pass_id uniquely identifies a run of an HLO pass on686 // that module. Note that a pass_id may not always refer to the same pass690 int64 pass_id = 1; field
1815 void set_creation_pass_id(int64_t pass_id) { in set_creation_pass_id() argument1816 metadata_.set_creation_pass_id(pass_id); in set_creation_pass_id()1821 void set_logical_creation_pass_id(int64_t pass_id) { in set_logical_creation_pass_id() argument1822 metadata_.set_logical_creation_pass_id(pass_id); in set_logical_creation_pass_id()
389 EXPECT_NE(pass_metadata.pass_id(), 0); in TEST_F()
303 // HloPassMetadata.pass_id of the pass that created this HLO instruction308 // HloPassMetadata.pass_id of the pass that created the logical functionality