(
self, summary_collection: dict[str, Any], name: str, value: Union[Tensor, WeightedSummary]
)
| 468 | assert_allclose(ref_per_example_loss, per_example_loss * per_example_weight) |
| 469 | |
| 470 | def _check_summary( |
| 471 | self, summary_collection: dict[str, Any], name: str, value: Union[Tensor, WeightedSummary] |
| 472 | ): |
| 473 | self.assertIn(name, summary_collection) |
| 474 | msg = f"mismatch in {name}: {summary_collection[name]} vs {value}" |
| 475 | self.assertEqual(summary_collection[name], value, msg) |
| 476 | |
| 477 | @set_threefry_partitionable(False) # TODO(yongqiang): update for threefry_partitionable True |
| 478 | def test_forward_summary(self): |
no outgoing calls
no test coverage detected