Searched refs:add_fn (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/aot/tests/ |
| D | tfcompile_test.cc | 435 FunctionComp add_fn; in TEST() local
436 EXPECT_EQ(add_fn.arg0_data(), add_fn.arg_data(0)); in TEST()
437 EXPECT_EQ(add_fn.arg1_data(), add_fn.arg_data(1)); in TEST()
439 add_fn.arg0() = 1; in TEST()
440 add_fn.arg1() = 2; in TEST()
441 EXPECT_TRUE(add_fn.Run()); in TEST()
442 EXPECT_EQ(add_fn.error_msg(), ""); in TEST()
443 EXPECT_EQ(add_fn.result0(), 3); in TEST()
444 EXPECT_EQ(add_fn.result0_data()[0], 3); in TEST()
445 EXPECT_EQ(add_fn.result0_data(), add_fn.results()[0]); in TEST()
|
| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | check_numerics_callback_test.py | 115 def add_fn(x, y): function
125 self.assertAllClose(self.evaluate(add_fn(x, y)), 5.0)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry.h | 155 const VariantBinaryOpFn& add_fn) { in RegisterBinaryOpFn() argument
163 {op, GetPersistentStringPiece(device), type_index}, add_fn)); in RegisterBinaryOpFn()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/ |
| D | np_interop_test.py | 404 add_fn = tf.function(add)
407 fn_result = add_fn(x, y)
|
| /external/llvm-project/llvm/test/Transforms/LowerMatrixIntrinsics/ |
| D | remarks-inlining.ll | 127 !21 = distinct !DISubprogram(name: "add_fn", scope: !1, file: !20, line: 1, type: !6, isLocal: fals…
|