| /external/tensorflow/tensorflow/core/framework/ |
| function_testlib.h | 88 FunctionDef XTimesFour();
|
| function_test.cc | 1050 // Test adding a gradient (XTimesFour isn't a valid grad function for
1054 grad.set_gradient_func(test::function::XTimesFour().signature().name());
[all...] |
| function_testlib.cc | 282 FunctionDef XTimesFour() {
285 "XTimesFour",
312 {{"x4"}, "XTimesFour", {"x"}, {{"T", "$T"}}},
313 {{"y"}, "XTimesFour", {"x4:y:0"}, {{"T", "$T"}}},
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| lower_if_op_test.cc | 61 *(f_lib_proto.add_function()) = test::function::XTimesFour();
75 .Attr("else_branch", FuncAttr("XTimesFour"))
254 FunctionDef x_times_four = test::function::XTimesFour();
276 eb.mutable_func()->set_name("XTimesFour");
|
| function_test.cc | 390 Init({test::function::XTimesTwo(), test::function::XTimesFour(),
398 InstantiateAndRun(flr0_, "XTimesFour", {{"T", DT_FLOAT}}, {x}, {&y}));
409 *proto.add_function() = test::function::XTimesFour();
428 TF_CHECK_OK(InstantiateAndRun(flr0_, "XTimesFour", {{"T", DT_FLOAT}}, options,
447 xt4_override.mutable_signature()->set_name("XTimesFour");
449 // Call XTimesFour via PartitionedCall which delays functions instantiation
458 {"f", FDH::FunctionRef("XTimesFour", {{"T", DT_FLOAT}})}}}},
464 *lib.add_function() = test::function::XTimesFour();
479 // Overlay library that overrides default XTimesFour with XTimesTwo body.
486 // We should call the XTimesFour override which is actually x*2
[all...] |
| lower_if_while_test.cc | 80 else_func.set_name("XTimesFour");
132 *f_lib_proto.add_function() = test::function::XTimesFour();
|
| shape_refiner_test.cc | [all...] |
| process_function_library_runtime_test.cc | 300 Init({test::function::XTimesTwo(), test::function::XTimesFour()});
313 Run("XTimesFour", opts, {{"T", DT_FLOAT}}, instantiate_opts, {x}, {&y}));
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| implementation_selector_test.cc | 105 auto gpu_def = test::function::XTimesFour();
115 NDef("y", "XTimesFour", {"x"}, {{"T", DT_FLOAT}}, CpuDevice),
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_dataset_op_test.cc | 165 10, 0, -3, "XTimesFour", std::vector<int64>{40, 28, 16, 4},
167 test::function::XTimesFour()})));
528 RoundtripTestParams(0, 10, 2, 6, 32, "XTimesFour",
531 test::function::XTimesFour()})));
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_partition_test.cc | 464 *fdef_lib.add_function() = test::function::XTimesFour();
471 ConstructOp(in_.WithOpName("B2"), "XTimesFour", {b1});
484 // Node "B2" is placed in part `b`, and uses both "XTimesFour" directly,
485 // and "XTimesTwo" in the body of "XTimesFour".
486 ExpectFunctions(partitions_[b].library(), {"XTimesTwo", "XTimesFour"});
|
| graph_test.cc | 597 *proto.add_function() = test::function::XTimesFour();
600 EXPECT_TRUE(graph_.flib_def().Find("XTimesFour") != nullptr);
605 EXPECT_TRUE(graph_.flib_def().Find("XTimesFour") != nullptr);
|
| /external/tensorflow/tensorflow/core/grappler/ |
| mutable_graph_view_test.cc | 196 FunctionDef x_times_four = test::function::XTimesFour();
[all...] |
