| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | shuffle_and_repeat_fusion_test.cc | 40 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
41 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
42 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
52 graph_utils::AddScalarConstNode<int64>(128, &graph); in TEST()
53 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
54 NodeDef *seed2_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
63 NodeDef *count_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
108 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
109 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
110 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
[all …]
|
| D | map_and_batch_fusion_test.cc | 32 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
33 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
34 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
44 graph_utils::AddScalarConstNode<StringPiece>("hello", &graph); in TEST()
62 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST()
115 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
116 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
117 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
127 graph_utils::AddScalarConstNode<StringPiece>("hello", &graph); in TEST()
145 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST()
[all …]
|
| D | graph_utils.h | 61 NodeDef* AddScalarConstNode(T v, MutableGraphView* graph) { in AddScalarConstNode() function
70 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph);
72 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph);
74 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph);
76 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph);
78 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph);
80 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph);
|
| D | map_vectorization_test.cc | 76 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapNode()
107 auto buffer_size_node = graph_utils::AddScalarConstNode(buffer_size, graph); in AddPrefetchNode()
121 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode()
125 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddBatchNode()
148 auto start = graph_utils::AddScalarConstNode(static_cast<int64>(0), graph); in AddRangeNode()
149 auto stop = graph_utils::AddScalarConstNode(static_cast<int64>(10), graph); in AddRangeNode()
150 auto step = graph_utils::AddScalarConstNode(static_cast<int64>(1), graph); in AddRangeNode()
288 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode()
290 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapAndBatchNode()
291 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddMapAndBatchNode()
|
| D | graph_utils_test.cc | 43 NodeDef* bool_node = AddScalarConstNode<bool>(true, &graph); in TEST()
51 NodeDef* double_node = AddScalarConstNode<double>(3.14, &graph); in TEST()
59 NodeDef* float_node = AddScalarConstNode<float>(3.14, &graph); in TEST()
67 NodeDef* int_node = AddScalarConstNode<int>(42, &graph); in TEST()
75 NodeDef* int64_node = AddScalarConstNode<int64>(42, &graph); in TEST()
83 NodeDef* string_node = AddScalarConstNode<StringPiece>("hello", &graph); in TEST()
|
| D | map_and_batch_fusion.cc | 65 NodeDef* tmp = graph_utils::AddScalarConstNode<int64>( in MakeMapAndBatchNode()
69 NodeDef* tmp = graph_utils::AddScalarConstNode<int64>(1, graph); in MakeMapAndBatchNode()
77 NodeDef* tmp = graph_utils::AddScalarConstNode<bool>(false, graph); in MakeMapAndBatchNode()
|
| D | noop_elimination_test.cc | 40 NodeDef *node_count = graph_utils::AddScalarConstNode<int64>(count, graph); in MakeUnaryNode()
56 graph_utils::AddScalarConstNode<StringPiece>("", graph); in MakeCacheNode()
63 auto *start_node = graph_utils::AddScalarConstNode<int64>(0, graph); in MakeRangeNode()
64 auto *stop_node = graph_utils::AddScalarConstNode<int64>(10, graph); in MakeRangeNode()
65 auto *step_node = graph_utils::AddScalarConstNode<int64>(1, graph); in MakeRangeNode()
|
| D | graph_utils.cc | 123 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph) { in AddScalarConstNode() function
129 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph) { in AddScalarConstNode() function
135 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph) { in AddScalarConstNode() function
141 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph) { in AddScalarConstNode() function
147 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph) { in AddScalarConstNode() function
153 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph) { in AddScalarConstNode() function
|
| D | hoist_random_uniform.cc | 75 const auto* seed = graph_utils::AddScalarConstNode<int64>( in MakeRandomDataset()
77 const auto* seed2 = graph_utils::AddScalarConstNode<int64>( in MakeRandomDataset()
95 const auto* batch_size = graph_utils::AddScalarConstNode<int64>(2, graph); in MakeBatchTwo()
96 const auto* drop_reminder = graph_utils::AddScalarConstNode(false, graph); in MakeBatchTwo()
|
| D | auto_shard.cc | 115 graph_utils::AddScalarConstNode<int64>(num_workers, graph); in AddShardNode()
116 NodeDef* index_node = graph_utils::AddScalarConstNode<int64>(index, graph); in AddShardNode()
|
| D | map_parallelization.cc | 47 auto* num_parallel_calls = graph_utils::AddScalarConstNode(kAutotune, graph); in MakeParallelMap()
|
| D | latency_all_edges.cc | 45 NodeDef* tag = graph_utils::AddScalarConstNode<StringPiece>( in MakeLatencyNode()
|
| D | map_vectorization.cc | 275 graph_utils::AddScalarConstNode(static_cast<int32>(kAutotune), graph); in AddNewMapNode()
309 graph_utils::AddScalarConstNode(static_cast<int64>(kAutotune), graph); in AddNewPrefetchNode()
409 graph_utils::AddScalarConstNode(static_cast<int64>(1), graph); in AddNewChooseFastestNode()
|
| D | rebatch.cc | 166 graph_utils::AddScalarConstNode<int64>(batch_size, graph); in MutateBatchSize()
|
| D | fusion_utils.cc | 265 auto* node = graph_utils::AddScalarConstNode(false, &graph_view); in CreateFalsePredicate()
|