Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_metadata.cc38 TF_ASSIGN_OR_RETURN(HloPassMetadata * pass_metadata, in MutateCurrentHloPassMetadata()
40 mutator(pass_metadata); in MutateCurrentHloPassMetadata()
45 HloPassMetadata* pass_metadata = module_metadata_.add_pass_metadata(); in RecordPassStart() local
46 pass_metadata->set_pass_id(next_pass_id_++); in RecordPassStart()
47 pass_metadata->set_start_timestamp_usec(env_->NowMicros()); in RecordPassStart()
48 running_passes_.push_back(pass_metadata); in RecordPassStart()
52 TF_ASSIGN_OR_RETURN(HloPassMetadata * pass_metadata, in RecordPassEnd()
54 pass_metadata->set_end_timestamp_usec(env_->NowMicros()); in RecordPassEnd()
71 for (const HloPassMetadata& pass_metadata : in set_prepartitioning_metadata() local
72 prepartitioning_metadata.proto().pass_metadata()) { in set_prepartitioning_metadata()
[all …]
Dhlo_module_metadata.h65 TF_ASSIGN_OR_RETURN(HloPassMetadata * pass_metadata, in current_pass_id()
67 return pass_metadata->pass_id(); in current_pass_id()
73 [&pass_name](HloPassMetadata* pass_metadata) { in set_current_pass_name()
74 pass_metadata->set_pass_name(pass_name); in set_current_pass_name()
79 [&pipeline_name](HloPassMetadata* pass_metadata) { in set_current_pass_pipeline_name()
80 pass_metadata->set_pipeline_name(pipeline_name); in set_current_pass_pipeline_name()
85 [&dump_filename](HloPassMetadata* pass_metadata) { in add_current_pass_dump_filename()
86 pass_metadata->add_dump_filenames(dump_filename); in add_current_pass_dump_filename()
91 [&module_changed](HloPassMetadata* pass_metadata) { in set_current_pass_module_changed()
92 pass_metadata->set_module_changed(module_changed); in set_current_pass_module_changed()
[all …]
Dhlo_pass_pipeline_test.cc282 ASSERT_THAT(metadata.pass_metadata(), SizeIs(3)); in TEST_F()
283 for (int pass = 0; pass < metadata.pass_metadata().size(); pass++) { in TEST_F()
284 const HloPassMetadata& pass_metadata = metadata.pass_metadata(pass); in TEST_F() local
285 EXPECT_NE(pass_metadata.pass_id(), 0); in TEST_F()
286 EXPECT_THAT(pass_metadata.pass_name(), StrEq(pass_names[pass])); in TEST_F()
287 EXPECT_THAT(pass_metadata.pipeline_name(), StrEq(pipeline_name)); in TEST_F()
288 EXPECT_FALSE(pass_metadata.module_changed()); in TEST_F()
289 EXPECT_EQ(pass_metadata.module_id(), module->unique_id()); in TEST_F()
290 EXPECT_THAT(pass_metadata.module_group_module_ids(), in TEST_F()
293 EXPECT_GT(pass_metadata.start_timestamp_usec(), 0); in TEST_F()
[all …]
Dhlo_module_metadata_test.cc47 module_metadata.proto().pass_metadata(), in TEST()
58 module_metadata.proto().pass_metadata(), in TEST()
69 EXPECT_THAT(module_metadata.proto().pass_metadata(), in TEST()
75 EXPECT_THAT(module_metadata.proto().pass_metadata(), in TEST()
96 module_metadata.proto().pass_metadata(), in TEST()
106 module_metadata.proto().pass_metadata(), in TEST()
134 new_module_metadata.proto().pass_metadata(), in TEST()
139 EXPECT_THAT(new_module_metadata.prepartitioning_metadata()->pass_metadata(), in TEST()
Dhlo_module_group_test.cc226 const HloPassMetadata& pass_metadata = in TEST_F() local
227 *module_metadata.pass_metadata().rbegin(); in TEST_F()
228 EXPECT_THAT(pass_metadata, in TEST_F()
Dhlo.proto554 repeated HloPassMetadata pass_metadata = 5; field