| /external/tensorflow/tensorflow/core/common_runtime/ |
| optimization_registry_test.cc | 60 *func_def_lib.add_function() = test::function::XTimesTwo();
83 auto f1 = GetFunctionDef("XTimesTwo");
85 EXPECT_EQ(test::function::XTimesTwo().DebugString(), f1->DebugString());
|
| lower_if_while_test.cc | 78 then_func.set_name("XTimesTwo");
131 *f_lib_proto.add_function() = test::function::XTimesTwo();
199 body_func.set_name("XTimesTwo");
241 *f_lib_proto.add_function() = test::function::XTimesTwo();
253 else_func.mutable_func()->set_name("XTimesTwo");
|
| function_test.cc | 125 TEST_F(FunctionTest, XTimesTwo) {
126 Create(test::function::XTimesTwo(), {{"T", DT_FLOAT}});
377 TEST_F(FunctionLibraryRuntimeTest, XTimesTwo) {
378 Init({test::function::XTimesTwo()});
382 InstantiateAndRun(flr0_, "XTimesTwo", {{"T", DT_FLOAT}}, {x}, {&y}));
385 flr0_, "XTimesTwo", {{"T", DT_FLOAT}}, {x}, {&y}));
390 Init({test::function::XTimesTwo(), test::function::XTimesFour(),
395 InstantiateAndRun(flr0_, "XTimesTwo", {{"T", DT_FLOAT}}, {x}, {&y}));
408 *proto.add_function() = test::function::XTimesTwo();
421 HasError(InstantiateAndRun(flr0_, "XTimesTwo", {{"T", DT_FLOAT}}
[all...] |
| lower_if_op_test.cc | 60 *(f_lib_proto.add_function()) = test::function::XTimesTwo();
74 .Attr("then_branch", FuncAttr("XTimesTwo"))
253 FunctionDef x_times_two = test::function::XTimesTwo();
274 tb.mutable_func()->set_name("XTimesTwo");
|
| lower_while_op_test.cc | 52 *f_lib_proto.add_function() = test::function::XTimesTwo();
63 body_func.mutable_func()->set_name("XTimesTwo");
248 FunctionDef x_times_two = test::function::XTimesTwo();
268 body_func.mutable_func()->set_name("XTimesTwo");
|
| function_threadpool_test.cc | 203 Init({test::function::BlockingOpFn(), test::function::XTimesTwo()}, tp);
221 TF_CHECK_OK(Instantiate(flr0_, "XTimesTwo", {{"T", DT_FLOAT}}, &h));
|
| process_function_library_runtime_test.cc | 268 Init({test::function::XTimesTwo()});
278 Run("XTimesTwo", opts, {{"T", DT_FLOAT}}, instantiate_opts, {x}, {&y}));
300 Init({test::function::XTimesTwo(), test::function::XTimesFour()});
310 Run("XTimesTwo", opts, {{"T", DT_FLOAT}}, instantiate_opts, {x}, {&y}));
|
| shape_refiner_test.cc | [all...] |
| constant_folding_test.cc | 338 *flib.add_function() = test::function::XTimesTwo();
351 NodeDefBuilder("times_two", "XTimesTwo", s.graph()->op_registry())
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| map_fusion_test.cc | 45 test::function::XTimesTwo(),
70 test::function::XTimesTwo(),
|
| map_parallelization_test.cc | 32 const char stateless_fun_name[] = "XTimesTwo";
46 test::function::XTimesTwo(),
71 test::function::XTimesTwo(),
|
| make_numa_aware_test.cc | 52 test::function::XTimesTwo(),
87 test::function::XTimesTwo(),
|
| make_sloppy_test.cc | 50 "XTimesTwo", /*sloppy=*/false)},
53 test::function::XTimesTwo(),
74 MakeParallelMapNode("map", "range", "num_parallel_calls", "XTimesTwo",
78 test::function::XTimesTwo(),
|
| fusion_utils_test.cc | 57 *parent_function = test::function::XTimesTwo();
59 *function = test::function::XTimesTwo();
95 *xtimes_two = test::function::XTimesTwo();
135 *parent_function = test::function::XTimesTwo();
137 *function = test::function::XTimesTwo();
153 *function = test::function::XTimesTwo();
|
| function_utils_test.cc | 57 FunctionDef function = test::function::XTimesTwo();
64 FunctionDef function = test::function::XTimesTwo();
71 FunctionDef function = test::function::XTimesTwo();
78 FunctionDef function = test::function::XTimesTwo();
84 FunctionDef function = test::function::XTimesTwo();
92 FunctionDef function = test::function::XTimesTwo();
100 FunctionDef function = test::function::XTimesTwo();
106 FunctionDef function = test::function::XTimesTwo();
112 FunctionDef function = test::function::XTimesTwo();
790 *function = test::function::XTimesTwo();
[all...] |
| map_and_filter_fusion_test.cc | 46 test::function::XTimesTwo(),
72 MakeParallelMapNode("map", "range", "num_parallel_calls", "XTimesTwo",
77 test::function::XTimesTwo(),
110 test::function::XTimesTwo(),
153 MakeParallelMapNode("map", "range", "num_parallel_calls", "XTimesTwo",
159 test::function::XTimesTwo(),
|
| /external/tensorflow/tensorflow/core/framework/ |
| function_test.cc | 391 TEST(TFunc, XTimesTwo) {
393 XTimesTwo[T:{float, double, int32, int64}](x:T) -> (y:T) {
400 EXPECT_EQ(expect, DebugString(test::function::XTimesTwo()));
971 *proto.add_function() = test::function::XTimesTwo();
977 XTimesTwo[T:{float, double, int32, int64}](x:T) -> (y:T) {
984 auto found = lib_def.Find("XTimesTwo");
991 *proto.add_function() = test::function::XTimesTwo();
997 TF_EXPECT_OK(lib_def.LookUpOpDef("XTimesTwo", &op_def));
1000 test::function::XTimesTwo().signature().DebugString());
1003 TF_EXPECT_OK(lib_def.LookUp("XTimesTwo", &op_reg_data))
[all...] |
| function_testlib.h | 64 FunctionDef XTimesTwo();
|
| function_testlib.cc | 119 FunctionDef XTimesTwo() {
123 "XTimesTwo",
294 {{"x2"}, "XTimesTwo", {"x"}, {{"T", "$T"}}},
295 {{"y"}, "XTimesTwo", {"x2:y:0"}, {{"T", "$T"}}},
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| map_dataset_op_test.cc | 160 0, 10, 3, "XTimesTwo", std::vector<int64>{0, 6, 12, 18},
161 std::vector<FunctionDef>{test::function::XTimesTwo()}),
166 std::vector<FunctionDef>{test::function::XTimesTwo(),
172 FunctionDef func_def = test::function::XTimesTwo();
199 FunctionDef func_def = test::function::XTimesTwo();
227 FunctionDef func_def = test::function::XTimesTwo();
278 FunctionDef func_def = test::function::XTimesTwo();
311 FunctionDef func_def = test::function::XTimesTwo();
343 FunctionDef func_def = test::function::XTimesTwo();
376 FunctionDef func_def = test::function::XTimesTwo();
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| implementation_selector_test.cc | 62 auto cpu_def = test::function::XTimesTwo();
77 NDef("y1", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, GpuDevice),
79 NDef("y2", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, CpuDevice),
93 EXPECT_EQ("XTimesTwo", node.op());
100 auto cpu_def = test::function::XTimesTwo();
|
| function_optimizer_test.cc | 48 // Build a graph to compute y = XTimesTwo(x)
52 NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice),
56 test::function::XTimesTwo(),
123 // Standard XTimesTwo() function.
124 FunctionDef x_times_two = test::function::XTimesTwo();
146 NDef("y1", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice),
160 // And that XTimesTwo was successfully inlined.
199 // Create and instantiate a version of the XTimesTwo function that only
204 "XTimesTwo",
226 NDef("y", "XTimesTwo", {"x"}, {}, kDevice)
[all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass_test.cc | 577 *library_in.add_function() = test::function::XTimesTwo();
594 *library.add_function() = test::function::XTimesTwo();
613 *library_expected.add_function() = test::function::XTimesTwo();
647 *library.add_function() = test::function::XTimesTwo();
668 *library_expected.add_function() = test::function::XTimesTwo();
[all...] |
| /external/tensorflow/tensorflow/core/grappler/ |
| grappler_item_builder_test.cc | 170 // y = XTimesTwo(x)
174 test::function::NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}},
178 test::function::XTimesTwo(),
194 // y = XTimesTwo(x)
|
| mutable_graph_view_test.cc | 176 FunctionDef x_times_two = test::function::XTimesTwo();
184 FunctionDef x_times_two = test::function::XTimesTwo();
197 x_times_four.mutable_signature()->set_name("XTimesTwo");
202 FunctionDef x_times_two = test::function::XTimesTwo();
209 "different function definition with the same name: XTimesTwo.");
[all...] |
