Home
last modified time | relevance | path

Searched refs:NDef (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc44 using test::function::NDef; in TEST_F()
51 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
52 NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
53 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
119 using test::function::NDef; in TEST_F()
145 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
146 NDef("y1", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
147 NDef("y2", "MyIdentityN", {"x"}, {{"T", DT_FLOAT}, {"N", 1}}, kDevice), in TEST_F()
148 NDef("z1", "Identity", {"y1:0"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
149 NDef("z2", "Identity", {"y2:0"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
[all …]
Dmeta_optimizer_test.cc236 using test::function::NDef; in TEST_F()
289 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
290 NDef("b", "Placeholder", {}, {{"dtype", DT_INT32}}, kDevice), in TEST_F()
292 NDef("square", "MySquare", {"a"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
293 NDef("quadratic", "MyQuadratic", {"b"}, {{"T", DT_INT32}}, kDevice), in TEST_F()
295 NDef("out_s", "Identity", {"square:0"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
296 NDef("out_q", "Identity", {"quadratic:0"}, {{"T", DT_INT32}}, kDevice)}, in TEST_F()
408 using test::function::NDef; in TEST_F()
445 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
446 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
[all …]
Dimplementation_selector_test.cc61 using test::function::NDef; in TEST_F()
76 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, GpuDevice), in TEST_F()
77 NDef("y1", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, GpuDevice), in TEST_F()
78 NDef("z1", "Identity", {"y1"}, {{"T", DT_FLOAT}}, GpuDevice), in TEST_F()
79 NDef("y2", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, CpuDevice), in TEST_F()
80 NDef("z2", "Identity", {"y2"}, {{"T", DT_FLOAT}}, CpuDevice)}, in TEST_F()
99 using test::function::NDef; in TEST_F()
114 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, CpuDevice), in TEST_F()
115 NDef("y", "XTimesFour", {"x"}, {{"T", DT_FLOAT}}, CpuDevice), in TEST_F()
116 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, CpuDevice)}, in TEST_F()
[all …]
Dgraph_optimizer_stage_test.cc30 using ::tensorflow::test::function::NDef;
87 GDef({NDef("a/b/c/A", "NotImportant", {}), in TEST_F()
88 NDef("a/b/c/my_opt/my_stg_A", "NotImportant", {}), in TEST_F()
89 NDef("a/b/c/my_opt/my_stg_my_rewrite_A", "NotImportant", {})}, in TEST_F()
114 {NDef("a/b/c/A", "NotImportant", {}), in TEST_F()
115 NDef("a/b/c/my_opt/my_stg_A", "NotImportant", {}), in TEST_F()
116 NDef("a/b/c/my_opt/my_stg_A_unique0", "NotImportant", {}), in TEST_F()
117 NDef("a/b/c/my_opt/my_stg_my_rewrite_A", "NotImportant", {}), in TEST_F()
118 NDef("a/b/c/my_opt/my_stg_my_rewrite_A_unique1", "NotImportant", {})}, in TEST_F()
/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal_test.cc26 using ::tensorflow::test::function::NDef;
42 {NDef("2", op, {"5"}, {}), // in TEST()
43 NDef("0", op, {"5", "4"}, {}), // in TEST()
44 NDef("1", op, {"4", "3"}, {}), // in TEST()
45 NDef("3", op, {"2"}, {}), // in TEST()
46 NDef("5", op, {}, {}), // in TEST()
47 NDef("4", op, {}, {})}, // in TEST()
73 {NDef("2", op, {"5"}, {}), // in TEST()
74 NDef("0", op, {"5", "4"}, {}), // in TEST()
75 NDef("1", op, {"4", "3"}, {}), // in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/
Dmutable_graph_view_test.cc33 using ::tensorflow::test::function::NDef;
147 NDef("foo", "NotImportant", {}, {}), in TEST()
148 NDef("bar", "NotImportant", {}, {}), in TEST()
149 NDef("baz", "NotImportant", {"foo", "bar"}), in TEST()
158 NDef("s/n0", "NotImportant", {}, {}), in TEST()
159 NDef("s/n1", "NotImportant", {"bar", "s/n0"}, {}), in TEST()
215 {NDef("bar_1", "Switch", {}, {}), NDef("bar_2", "Identity", {"bar_1:1"}), in TEST()
216 NDef("other", "NotImportant", {}, {}), in TEST()
217 NDef("foo_1", "NotImportant", {"bar_2", "other", "bar_2:1", "^bar_2"}), in TEST()
218 NDef("foo_2", "NotImportant", {"other:1", "bar_2:2", "^bar_2"})}, in TEST()
[all …]
Dgrappler_item_test.cc50 using test::function::NDef; in TEST_F()
59 NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, cpu0), in TEST_F()
60 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, cpu1), in TEST_F()
61 NDef("c", "Placeholder", {}, {{"dtype", DT_FLOAT}}, cpu2), in TEST_F()
Dgrappler_item_builder_test.cc173 {test::function::NDef("x", "Const", {}, {{"dtype", DT_FLOAT}}, device), in TEST_F()
174 test::function::NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, in TEST_F()
197 {test::function::NDef("x", "Const", {}, {{"dtype", DT_FLOAT}}, device), in TEST_F()
198 test::function::NDef("y", "CustomOp", {"x"}, {{"T", DT_FLOAT}}, device)}, in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmake_sloppy_test.cc37 using test::function::NDef; in TEST()
40 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
41 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
42 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
43 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
44 NDef("cycle_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
45 NDef("block_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
46 NDef("num_parallel_calls", "Const", {}, in TEST()
65 using test::function::NDef; in TEST()
68 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
[all …]
Dmap_and_filter_fusion_test.cc36 using test::function::NDef; in TEST()
39 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
40 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
41 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
42 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
63 using test::function::NDef; in TEST()
66 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
67 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
68 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
69 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
[all …]
Dmake_numa_aware_test.cc33 using test::function::NDef; in TEST()
37 NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
38 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
39 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
40 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
41 NDef("batch_size", "Const", {}, {{"value", 3}, {"dtype", DT_INT32}}), in TEST()
42 NDef("num_parallel_calls", "Const", {}, in TEST()
44 NDef("drop_remainder", "Const", {}, in TEST()
67 using test::function::NDef; in TEST()
71 NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
[all …]
Dfilter_fusion_test.cc34 using test::function::NDef; in TEST()
37 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
38 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
39 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
40 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
57 using test::function::NDef; in TEST()
60 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
61 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
62 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
63 NDef("filename", "Const", {}, {{"value", ""}, {"dtype", DT_STRING}}), in TEST()
[all …]
Dmap_parallelization_test.cc36 using test::function::NDef; in TEST()
39 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
40 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
41 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
42 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
58 using test::function::NDef; in TEST()
61 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
62 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
63 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
64 NDef("filename", "Const", {}, {{"value", ""}, {"dtype", DT_STRING}}), in TEST()
[all …]
Dmap_fusion_test.cc35 using test::function::NDef; in TEST()
38 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
39 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
40 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
41 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
57 using test::function::NDef; in TEST()
60 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
61 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
62 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
63 NDef("filename", "Const", {}, {{"value", ""}, {"dtype", DT_STRING}}), in TEST()
[all …]
Dhoist_random_uniform_test.cc32 using test::function::NDef; in TEST()
35 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
36 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
37 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
38 NDef("filename", "Const", {}, {{"value", ""}, {"dtype", DT_STRING}}), in TEST()
39 NDef("range", "RangeDataset", {"start", "stop", "step"}, in TEST()
43 NDef("cache", "CacheDataset", {"map1", "filename"}, {})}, in TEST()
Dlatency_all_edges_test.cc30 using test::function::NDef; in TEST()
33 NDef("component_node", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}); in TEST()
35 NDef("from_tensor_node", "TensorDataset", {"component_node"}, in TEST()
38 NodeDef captured_input_node = NDef("captured_input_node", "Const", {}, in TEST()
40 NodeDef map_node = NDef("map_node", "MapDataset", in TEST()
46 NodeDef buffer_size_node = NDef("buffer_size_node", "Const", {}, in TEST()
48 NodeDef prefetch_node = NDef("prefetch_node", "Prefetch_Dataset", in TEST()
Dgraph_test_utils.cc30 return test::function::NDef( in MakeFilterNode()
43 return test::function::NDef( in MakeMapAndBatchNode()
55 return test::function::NDef( in MakeMapNode()
69 return test::function::NDef( in MakeParallelInterleaveNode()
85 return test::function::NDef( in MakeParallelMapNode()
100 return test::function::NDef( in MakeParseExampleNode()
Dmap_vectorization_test.cc54 using test::function::NDef;
78 NDef(/*name=*/"map", /*op=*/kParallelMapOp, in AddMapNode()
90 NDef(/*name=*/"map", /*op=*/kMapOp, in AddMapNode()
109 NDef(/*name=*/"prefetch", /*op=*/kPrefetchOp, in AddPrefetchNode()
126 result = NDef( in AddBatchNode()
135 NDef(/*name=*/"batch", /*op=*/kBatchOp, in AddBatchNode()
153 NDef(/*name=*/"range", /*op=*/kRangeOp, in AddRangeNode()
294 NDef(/*name=*/"map_and_batch", in AddMapAndBatchNode()
417 NodeDef result = NDef(/*name=*/"input", /*op=*/"InputDataset", in AddArbitraryInputNode()
/external/tensorflow/tensorflow/core/ops/
Darray_grad_test.cc40 {f::NDef("x0", "Placeholder", {}, {{"dtype", T}}), in PackGrad()
41 f::NDef("x1", "Placeholder", {}, {{"dtype", T}}), in PackGrad()
42 f::NDef("axis", "Placeholder", {}, {{"dtype", DT_INT32}}), in PackGrad()
43 f::NDef("dy", "Placeholder", {}, {{"dtype", T}}), in PackGrad()
44 f::NDef("dx", "SymbolicGradient", {"x0", "x1", "dy"}, in PackGrad()
81 {f::NDef("x", "Placeholder", {}, {{"dtype", T}}), in UnpackGrad()
82 f::NDef("axis", "Placeholder", {}, {{"dtype", DT_INT32}}), in UnpackGrad()
83 f::NDef("dy0", "Placeholder", {}, {{"dtype", T}}), in UnpackGrad()
84 f::NDef("dy1", "Placeholder", {}, {{"dtype", T}}), in UnpackGrad()
85 f::NDef("dx", "SymbolicGradient", {"x", "dy0", "dy1"}, in UnpackGrad()
[all …]
Dmath_grad_test.cc79 f::NDef("x", "Placeholder", {}, {{"dtype", src}}), in Unary()
80 f::NDef("dx", "TestGrad", {"x"}, {}), in Unary()
155 f::NDef("x", "Placeholder", {}, {{"dtype", T}}), in SymGrad()
156 f::NDef("y", "Placeholder", {}, {{"dtype", T}}), in SymGrad()
157 f::NDef("d", "TestGrad", {"x", "y"}, {}), in SymGrad()
210 f::NDef("x", "Placeholder", {}, {{"dtype", T}}), in ReductionGrad()
211 f::NDef("i", "Placeholder", {}, {{"dtype", DT_INT32}}), in ReductionGrad()
212 f::NDef("d", "TestGrad", {"x", "i"}, {}), in ReductionGrad()
233 f::NDef("x", "Placeholder", {}, {{"dtype", T}}), in MatMulCommon()
234 f::NDef("y", "Placeholder", {}, {{"dtype", T}}), in MatMulCommon()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc49 using ::tensorflow::test::function::NDef;
1696 NDef("a", "_Arg", {}, {{"T", DT_RESOURCE}}, kGPU), in TEST_P()
1697 NDef("b", "_Arg", {}, {{"T", DT_RESOURCE}}, kCPU), in TEST_P()
1698 NDef("id1", "Identity", {"a"}, in TEST_P()
1701 NDef("id2", "Identity", {"b"}, {{"T", DT_RESOURCE}}), in TEST_P()
1741 NDef("a", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST_F()
1742 NDef("b", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST_F()
1743 NDef("id_a", "Identity", {"a"}, {{"T", DT_RESOURCE}}, kGPU), in TEST_F()
1744 NDef("id_b", "Identity", {"b"}, {{"T", DT_RESOURCE}}, kCPU), in TEST_F()
1745 NDef("id1", "Identity", {"id_a"}, in TEST_F()
[all …]
Dfunction_test.cc890 using test::function::NDef; in TEST_F()
941 {NDef("a", "_Arg", {}, {{"T", DT_FLOAT}, {"index", 0}}), in TEST_F()
942 NDef(input_node, "Identity", {"a"}, {{"T", DT_FLOAT}}), in TEST_F()
943 NDef("b/add", "Add", {input_node, input_node}, {{"T", DT_FLOAT}}), in TEST_F()
944 NDef("b/ret", "Mul", {input_node, input_node}, {{"T", DT_FLOAT}}), in TEST_F()
945 NDef(output_node, "Identity", {"b/ret"}, {{"T", DT_FLOAT}}), in TEST_F()
946 NDef(output_control_node, "NoOp", {"^Func/b/output/_1"}, {}), in TEST_F()
947 NDef("c", "NoOp", {"^" + output_control_node}, {}), in TEST_F()
948 NDef("ret", "_Retval", {output_node, "^c"}, in TEST_F()
964 {NDef("a", "_Arg", {}, {{"T", DT_FLOAT}, {"index", 0}}), in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/framework/
Dfunction_test.cc1352 using ::tensorflow::test::function::NDef; in TEST()
1383 NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1384 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST()
1385 NDef("x", "Func1", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
1386 NDef("y", "PartitionedCall", {"a", "b"}, in TEST()
1391 NDef("z", "Func4", {"a", "b"}, {{"T", DT_FLOAT}}, kDevice), in TEST()
Dfunction_testlib.h50 NodeDef NDef(
Dfunction_testlib.cc48 NodeDef NDef(StringPiece name, StringPiece op, gtl::ArraySlice<string> inputs, in NDef() function

12