Searched refs:sig_def (Results 1 – 2 of 2) sorted by relevance
345 sig_def = outputter.as_signature_def(receiver)347 self.assertIn('loss/my_loss', sig_def.outputs)348 self.assertIn('metrics_1/value', sig_def.outputs)349 self.assertIn('metrics_2/value', sig_def.outputs)350 self.assertIn('predictions/output1', sig_def.outputs)351 self.assertIn('features', sig_def.inputs)362 sig_def = outputter.as_signature_def(receiver)364 self.assertIn('loss/my_loss', sig_def.outputs)365 self.assertNotIn('metrics/value', sig_def.outputs)366 self.assertIn('predictions/output1', sig_def.outputs)[all …]
297 for (const auto& sig_def : signature_defs_) { in signature_def_names() local298 method_names.emplace_back(&sig_def.method_name); in signature_def_names()309 for (const auto& sig_def : signature_defs_) { in signature_inputs() local310 if (sig_def.method_name == method_name) return sig_def.inputs; in signature_inputs()323 for (const auto& sig_def : signature_defs_) { in signature_outputs() local324 if (sig_def.method_name == method_name) return sig_def.outputs; in signature_outputs()