Searched refs:RecordPassEnd (Results 1 – 4 of 4) sorted by relevance
56 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()68 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()74 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()82 EXPECT_EQ(module_metadata.RecordPassEnd().code(), in TEST()86 EXPECT_IS_OK(module_metadata.RecordPassEnd()); in TEST()87 EXPECT_EQ(module_metadata.RecordPassEnd().code(), in TEST()124 EXPECT_IS_OK(old_module_metadata.RecordPassEnd()); in TEST()
51 Status HloModuleMetadata::RecordPassEnd() { in RecordPassEnd() function in xla::HloModuleMetadata
44 Status RecordPassEnd();
61 TF_RETURN_IF_ERROR(module.metadata()->RecordPassEnd()); in AttemptRecordPassEndMetadata()