48   std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>(  in TEST()
50   std::vector<TensorSpec> OutputSpecs{  in TEST()
51       TensorSpec::createSpec<float>("StatefulPartitionedCall", {1})};  in TEST()
89   std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>(  in TEST()
91   std::vector<TensorSpec> OutputSpecs{  in TEST()
92       TensorSpec::createSpec<float>("StatefulPartitionedCall", {1})};  in TEST()
109   std::vector<TensorSpec> InputSpecs{  in TEST()
110       TensorSpec::createSpec<int32_t>("serving_default_input_1",  in TEST()
112       TensorSpec::createSpec<float>("this_feature_does_not_exist", {2, 5})};  in TEST()
118           TensorSpec::createSpec<float>("StatefulPartitionedCall", {1}),  in TEST()
152       {TensorSpec::createSpec<float>("the_float", {2, 3}), None});  in TEST()
153   Features.push_back({TensorSpec::createSpec<int64_t>("the_int", {2}),  in TEST()
156   auto Rewards = TensorSpec::createSpec<float>("reward", {1});  in TEST()
188       {TensorSpec::createSpec<float>("the_float", {2, 3}), None});  in TEST()
189   Features.push_back({TensorSpec::createSpec<int32_t>("the_int", {2}),  in TEST()
192   auto Rewards = TensorSpec::createSpec<int32_t>("reward", {1});  in TEST()
224       {TensorSpec::createSpec<float>("the_float", {2, 3}), None});  in TEST()
225   Features.push_back({TensorSpec::createSpec<int64_t>("the_int", {2}),  in TEST()
228   auto Rewards = TensorSpec::createSpec<float>("reward", {1});  in TEST()
253   Features.push_back({TensorSpec::createSpec<float>("the_float", {1}), None});  in TEST()
254   Features.push_back({TensorSpec::createSpec<int64_t>("the_int", {1}), None});  in TEST()
256   auto Rewards = TensorSpec::createSpec<float>("reward", {1});  in TEST()
278   Features.push_back({TensorSpec::createSpec<float>("the_float", {1}), None});  in TEST()
279   Features.push_back({TensorSpec::createSpec<int64_t>("the_int", {1}), None});  in TEST()
281   auto Rewards = TensorSpec::createSpec<float>("reward", {1});  in TEST()