Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Disolate_placer_inspection_required_ops_pass_test.cc37 using ::tensorflow::test::function::NDef;
94 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
95 NDef("f", "PartitionedCall", {"x"}, in TEST()
99 NDef("y", "_Retval", {"f:0"}, {{"T", DT_RESOURCE}}), in TEST()
106 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
107 NDef("x_f", "Identity", {"x"}, {{"T", DT_RESOURCE}}), in TEST()
108 NDef("f", "PartitionedCall", {"x_f"}, in TEST()
112 NDef("f_y", "Identity", {"f:0"}, {{"T", DT_RESOURCE}}), in TEST()
113 NDef("y", "_Retval", {"f_y:0"}, {{"T", DT_RESOURCE}}), in TEST()
133 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
[all …]
Dplacer_inspection_required_ops_utils_test.cc33 using ::tensorflow::test::function::NDef;
70 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
71 NDef("f", "PartitionedCall", {"x"}, in TEST()
75 NDef("y", "_Retval", {"f:0"}, {{"T", DT_RESOURCE}}), in TEST()
97 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
98 NDef("f", "ResourceIdentity", {"x"}), in TEST()
99 NDef("y", "_Retval", {"f:0"}, {{"T", DT_RESOURCE}}), in TEST()
122 NDef("x", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST()
123 NDef("f", "PartitionedCall", {"x"}, in TEST()
127 NDef("y", "_Retval", {"f:0"}, {{"T", DT_FLOAT}}), in TEST()
Dplacer_test.cc52 using ::tensorflow::test::function::NDef;
1957 NDef("a", "_Arg", {}, {{"T", DT_RESOURCE}}, kGPU), in TEST_P()
1958 NDef("b", "_Arg", {}, {{"T", DT_RESOURCE}}, kCPU), in TEST_P()
1959 NDef("id1", "Identity", {"a"}, in TEST_P()
1962 NDef("id2", "Identity", {"b"}, {{"T", DT_RESOURCE}}), in TEST_P()
2002 NDef("a", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST_F()
2003 NDef("b", "_Arg", {}, {{"T", DT_RESOURCE}}), in TEST_F()
2004 NDef("id_a", "Identity", {"a"}, {{"T", DT_RESOURCE}}, kGPU), in TEST_F()
2005 NDef("id_b", "Identity", {"b"}, {{"T", DT_RESOURCE}}, kCPU), in TEST_F()
2006 NDef("id1", "Identity", {"id_a"}, in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer_test.cc39 using test::function::NDef; in TEST_F()
46 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
47 NDef("y", "XTimesTwo", {"x"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
48 NDef("z", "Identity", {"y"}, {{"T", DT_FLOAT}}, kDevice)}, in TEST_F()
62 {NDef("x", "Placeholder", {}, {{"dtype", DT_FLOAT}}), in TEST_F()
63 NDef(arg0, "Identity", {"x"}, {{"T", DT_FLOAT}}), in TEST_F()
64 NDef("y/two", "Const", {}, {{"dtype", DT_INT64}, {"value", kTwo}}), in TEST_F()
65 NDef("y/scale", "Cast", {"y/two"}, in TEST_F()
67 NDef("y/y", "Mul", {arg0, "y/scale"}, {{"T", DT_FLOAT}}), in TEST_F()
68 NDef(ret0, "Identity", {"y/y"}, {{"T", DT_FLOAT}}), in TEST_F()
[all …]
Dimplementation_selector_test.cc63 using test::function::NDef; in TEST_F()
71 {NDef("x", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
73 NDef("case", "Case", {"x"}, {{"T", DT_FLOAT}}, GpuDevice)}); in TEST_F()
87 using test::function::NDef; in TEST_F()
95 {NDef("x", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
97 NDef("case", "StatelessCase", {"x"}, {{"T", DT_FLOAT}}, GpuDevice)}); in TEST_F()
111 using test::function::NDef; in TEST_F()
120 {NDef("x", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
122 NDef("case", "Case", {"x"}, {{"T", DT_FLOAT}}, GpuDevice), in TEST_F()
123 NDef("y", "DeviceIndex", {}, {{"device_names", device_names}}, in TEST_F()
[all …]
Dmeta_optimizer_test.cc241 using test::function::NDef; in TEST_F()
294 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
295 NDef("b", "Placeholder", {}, {{"dtype", DT_INT32}}, kDevice), in TEST_F()
297 NDef("square", "MySquare", {"a"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
298 NDef("quadratic", "MyQuadratic", {"b"}, {{"T", DT_INT32}}, kDevice), in TEST_F()
300 NDef("out_s", "Identity", {"square:0"}, {{"T", DT_FLOAT}}, kDevice), in TEST_F()
301 NDef("out_q", "Identity", {"quadratic:0"}, {{"T", DT_INT32}}, kDevice)}, in TEST_F()
406 using test::function::NDef; in TEST_F()
443 {NDef("a", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
444 NDef("b", "Placeholder", {}, {{"dtype", DT_FLOAT}}, kDevice), in TEST_F()
[all …]
/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 …]
Dgraph_view_test.cc39 using ::tensorflow::test::function::NDef;
44 return GDef({NDef("a", kNoOp, {"b:2", "d:3", "b:2", "d:3", "^c"}), in SimpleTestGraph()
45 NDef("b", kNoOp, {"d:2", "c:5", "^c"}), in SimpleTestGraph()
46 NDef("c", kNoOp, {"^d", "^d"}), NDef("d", kNoOp, {})}, in SimpleTestGraph()
64 GDef({NDef("a", kNoOp, {}), NDef("a", kNoOp, {})}, /*funcs=*/{}); in TYPED_TEST()
76 GraphDef graph = GDef({NDef("a", kNoOp, {"b:3"})}, /*funcs=*/{}); in TYPED_TEST()
87 GraphDef graph = GDef({NDef("a", kNoOp, {"a:4"})}, /*funcs=*/{}); in TYPED_TEST()
99 GraphDef graph = GDef({NDef("a", kNoOp, {"^b", "b:4"}), NDef("b", kNoOp, {})}, in TYPED_TEST()
204 {NDef("a", kNoOp, {}), NDef("b", kNoOp, {}), NDef("c", kNoOp, {}), in TEST()
205 NDef("d", kNoOp, {"a:2", "b:1", "^c", "^c", "^a", "^a", "^b", "^c"})}, in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
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()
62 using test::function::NDef; in TEST()
65 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
66 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
67 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
68 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
[all …]
Dmake_sloppy_test.cc33 using test::function::NDef; in TEST()
36 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
37 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST()
38 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
39 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
40 NDef("cycle_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
41 NDef("block_length", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
42 NDef("num_parallel_calls", "Const", {}, in TEST()
61 using test::function::NDef; in TEST()
64 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
[all …]
Dbatch_parallelization_test.cc51 using test::function::NDef; in TEST_P()
54 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
55 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
56 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
57 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST_P()
58 NDef("batch_size", "Const", {}, {{"value", 2}, {"dtype", DT_INT32}}), in TEST_P()
59 NDef("drop_remainder", "Const", {}, in TEST_P()
62 NDef("Sink", "Identity", {"batch"}, {})}, in TEST_P()
83 using test::function::NDef; in TEST_P()
86 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
[all …]
Dparallel_batch_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("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
39 NDef("batch_size", "Const", {}, {{"value", 5}, {"dtype", DT_INT32}}), in TEST()
40 NDef("drop_remainder", "Const", {}, in TEST()
42 NDef("batch", "BatchDatasetV2", in TEST()
54 using test::function::NDef; in TEST()
57 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
[all …]
Dmap_parallelization_test.cc53 using test::function::NDef; in TEST_P()
56 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
57 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
58 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
59 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST_P()
61 NDef("Sink", "Identity", {"map"}, {})}, in TEST_P()
84 using test::function::NDef; in TEST_P()
87 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
88 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
89 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
[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_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 …]
Dautotune_buffer_sizes_test.cc48 using test::function::NDef; in TEST_P()
52 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
53 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
54 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
55 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST_P()
56 NDef("num_parallel_calls", "Const", {}, in TEST_P()
67 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
68 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
69 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
70 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST_P()
[all …]
Dreorder_data_discarding_ops_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("range", "RangeDataset", {"start", "stop", "step"}, in TEST()
44 NDef("take_count", "Const", {}, {{"value", 5}, {"dtype", DT_INT32}}), in TEST()
46 NDef("skip_count", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
48 NDef("batch_size", "Const", {}, {{"value", 2}, {"dtype", DT_INT32}}), in TEST()
49 NDef("drop_remainder", "Const", {}, in TEST()
54 NDef("num_shards", "Const", {}, {{"value", 2}, {"dtype", DT_INT32}}), in TEST()
[all …]
Dfilter_with_random_uniform_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", 10000}, {"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()
55 using test::function::NDef; in TEST()
58 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST()
59 NDef("stop", "Const", {}, {{"value", 10000}, {"dtype", DT_INT32}}), in TEST()
60 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST()
61 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST()
Denable_gradient_descent_test.cc52 using test::function::NDef; in TEST_P()
55 {NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
56 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
57 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
58 NDef("range", "RangeDataset", {"start", "stop", "step"}, {}), in TEST_P()
59 NDef("batch_size", "Const", {}, {{"value", 5}, {"dtype", DT_INT32}}), in TEST_P()
60 NDef("batch", "BatchDataset", {"range", "batch_size"}, {}), in TEST_P()
61 NDef("model", "ModelDataset", {"batch"}, in TEST_P()
63 NDef("Sink", op, {"model"}, {})}); in TEST_P()
Ddisable_prefetch_legacy_autotune_test.cc31 using test::function::NDef;
54 NDef("start", "Const", {}, {{"value", 0}, {"dtype", DT_INT32}}), in TEST_P()
55 NDef("stop", "Const", {}, {{"value", 10}, {"dtype", DT_INT32}}), in TEST_P()
56 NDef("step", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}), in TEST_P()
57 NDef("range", "RangeDataset", {"start", "stop", "step"}, in TEST_P()
60 NDef("prefetch1", "PrefetchDataset", {"range"}, in TEST_P()
62 NDef("prefetch2", "PrefetchDataset", {"prefetch1"}, in TEST_P()
64 NDef("prefetch3", "PrefetchDataset", {"prefetch2"}, {}), in TEST_P()
Dgraph_test_utils.cc31 return test::function::NDef( in MakeBatchV2Node()
43 return test::function::NDef( in MakeCacheV2Node()
58 return test::function::NDef( in MakeFilterNode()
71 return test::function::NDef( in MakeMapAndBatchNode()
83 return test::function::NDef( in MakeMapNode()
97 return test::function::NDef( in MakeParallelInterleaveV2Node()
113 return test::function::NDef( in MakeParallelMapNode()
128 return test::function::NDef( in MakeParseExampleNode()
141 return test::function::NDef( in MakeShuffleV2Node()
156 return test::function::NDef( in MakeTakeNode()
[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.cc33 using test::function::NDef; in TEST_P()
36 NDef("component_node", "Const", {}, {{"value", 1}, {"dtype", DT_INT32}}); in TEST_P()
38 NDef("from_tensor_node", "TensorDataset", {"component_node"}, in TEST_P()
41 NodeDef captured_input_node = NDef("captured_input_node", "Const", {}, in TEST_P()
43 NodeDef map_node = NDef("map_node", "MapDataset", in TEST_P()
49 NodeDef buffer_size_node = NDef("buffer_size_node", "Const", {}, in TEST_P()
52 NDef("prefetch_node", "PrefetchDataset", {"map_node", "buffer_size_node"}, in TEST_P()
54 NodeDef model_node = NDef("model_node", "ModelDataset", {"prefetch_node"}, in TEST_P()
/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 …]
/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 …]

12