| D | function_test.cc | 7 http://www.apache.org/licenses/LICENSE-2.0 68 return OpRegistry::Global()->LookUpOpDef(op, sig); in GetOpSig() 97 const int version = g->versions().producer(); in Create() 120 TF_CHECK_OK(exec_->Run(exec_args)); in Run() 137 auto x = test::AsTensor<float>({1, 2, 3, 4}); in TEST_F() local 139 Run({x}, {&y}); in TEST_F() 146 auto x = test::AsTensor<float>({1., 3., 2., 4.}, {2, 2}); in TEST_F() local 149 Run({w, x, b}, {&y}); in TEST_F() 159 device_count->insert({"CPU", 3}); in Init() 164 FunctionDefLibrary proto; in Init() local [all …]
|