Lines Matching refs:TensorSpec

55   const std::vector<TensorSpec> Inputs{  in TEST()
56       TensorSpec::createSpec<int64_t>("F1", {1}),  in TEST()
57       TensorSpec::createSpec<int64_t>("F2", {10}),  in TEST()
58       TensorSpec::createSpec<float>("F2", {5}),  in TEST()
76   std::vector<TensorSpec> Inputs{TensorSpec::createSpec<int64_t>("a", {1}),  in TEST()
77                                  TensorSpec::createSpec<int64_t>("b", {1})};  in TEST()
89   std::vector<TensorSpec> Inputs{TensorSpec::createSpec<int64_t>("a", {1}),  in TEST()
90                                  TensorSpec::createSpec<int64_t>("b", {1}),  in TEST()
91                                  TensorSpec::createSpec<int64_t>("c", {1})};  in TEST()
105   std::vector<TensorSpec> Inputs{  in TEST()
106       TensorSpec::createSpec<int64_t>("a", {1}),  in TEST()
107       TensorSpec::createSpec<int64_t>("c", {1}),  in TEST()
108       TensorSpec::createSpec<int64_t>("b", {1}),  in TEST()