Home
last modified time | relevance | path

Searched refs:add_fn (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/aot/tests/
Dtfcompile_test.cc413 FunctionComp add_fn; in TEST() local
414 EXPECT_EQ(add_fn.arg0_data(), add_fn.arg_data(0)); in TEST()
415 EXPECT_EQ(add_fn.arg1_data(), add_fn.arg_data(1)); in TEST()
417 add_fn.arg0() = 1; in TEST()
418 add_fn.arg1() = 2; in TEST()
419 EXPECT_TRUE(add_fn.Run()); in TEST()
420 EXPECT_EQ(add_fn.error_msg(), ""); in TEST()
421 EXPECT_EQ(add_fn.result0(), 3); in TEST()
422 EXPECT_EQ(add_fn.result0_data()[0], 3); in TEST()
423 EXPECT_EQ(add_fn.result0_data(), add_fn.results()[0]); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry.cc213 const VariantBinaryOpFn& add_fn) { in RegisterBinaryOpFn() argument
220 {op, GetPersistentStringPiece(device), type_index}, add_fn)); in RegisterBinaryOpFn()
Dvariant_op_registry.h125 const VariantBinaryOpFn& add_fn);