Searched refs:check_metadata (Results 1 – 14 of 14) sorted by relevance
77 auto check_metadata = [](const xla::OpSharding& sharding) { in TEST_P() local85 [&check_metadata]( in TEST_P()96 check_metadata(sharding_element); in TEST_P()99 check_metadata(sharding.value()); in TEST_P()
67 self.check_metadata()79 def check_metadata(self): member in check
58 def check_metadata(self): member in register
162 def check_metadata(self): member in sdist
70 self.check_metadata()82 def check_metadata(self): member in check
59 def check_metadata(self): member in register
170 def check_metadata(self): member in sdist
283 cmd.check_metadata()
287 cmd.check_metadata()
294 cmd.check_metadata()
275 cmd.check_metadata()
315 static void check_metadata(const expected_md* expected, in check_metadata() function354 check_metadata(state->expected, &state->md_array); in check_request_metadata()
2566 auto check_metadata = [](const HloSharding& sharding, in TEST_P() local2585 check_metadata(instruction->sharding(), CreateMetadata("a")); in TEST_P()2592 check_metadata(instruction->sharding(), CreateMetadata("b")); in TEST_P()
427 static void check_metadata(const expected_md* expected, in check_metadata() function466 check_metadata(state->expected, &state->md_array); in check_request_metadata()