Home
last modified time | relevance | path

Searched refs:AddScalarConstNode (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dshuffle_and_repeat_fusion_test.cc44 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
45 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
46 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
56 graph_utils::AddScalarConstNode<int64>(128, &graph); in TEST()
57 NodeDef *seed_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
58 NodeDef *seed2_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
68 NodeDef *count_node = graph_utils::AddScalarConstNode<int64>(-1, &graph); in TEST()
112 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
113 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
114 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
[all …]
Dmap_and_batch_fusion_test.cc32 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()
114 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST()
115 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST()
116 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST()
126 graph_utils::AddScalarConstNode<StringPiece>("hello", &graph); in TEST()
144 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST()
[all …]
Dgraph_utils.h62 NodeDef* AddScalarConstNode(T v, MutableGraphView* graph) { in AddScalarConstNode() function
71 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph);
73 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph);
75 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph);
77 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph);
79 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph);
81 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph);
Dgraph_utils_test.cc43 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()
91 NodeDef* int64_node = AddScalarConstNode<int64>(128, &graph); in TEST()
100 NodeDef* bool_node = AddScalarConstNode<bool>(true, &graph); in TEST()
120 NodeDef* int64_node = AddScalarConstNode<int64>(128, &graph); in TEST()
Dslack_test.cc42 NodeDef *start_node = graph_utils::AddScalarConstNode<int64>(0, &graph); in SetupGrapplerItem()
43 NodeDef *stop_node = graph_utils::AddScalarConstNode<int64>(10, &graph); in SetupGrapplerItem()
44 NodeDef *step_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in SetupGrapplerItem()
53 NodeDef *buffer_size_node = graph_utils::AddScalarConstNode<int64>(1, &graph); in SetupGrapplerItem()
Dautotune_buffer_sizes_test.cc161 NodeDef *start_val = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST_P()
162 NodeDef *stop_val = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST_P()
163 NodeDef *step_val = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST_P()
173 NodeDef *parallelism_val = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST_P()
188 graph_utils::AddScalarConstNode<int64>(initial_buffer_size, &graph); in TEST_P()
Dnoop_elimination_test.cc40 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()
Dmap_vectorization_test.cc89 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapNode()
120 auto buffer_size_node = graph_utils::AddScalarConstNode(buffer_size, graph); in AddPrefetchNode()
134 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode()
138 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddBatchNode()
161 auto start = graph_utils::AddScalarConstNode(static_cast<int64>(0), graph); in AddRangeNode()
162 auto stop = graph_utils::AddScalarConstNode(static_cast<int64>(10), graph); in AddRangeNode()
163 auto step = graph_utils::AddScalarConstNode(static_cast<int64>(1), graph); in AddRangeNode()
345 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode()
347 graph_utils::AddScalarConstNode(num_parallel_calls, graph); in AddMapAndBatchNode()
348 auto drop_remainder = graph_utils::AddScalarConstNode(true, graph); in AddMapAndBatchNode()
Dmap_and_batch_fusion.cc71 NodeDef* tmp = graph_utils::AddScalarConstNode<int64>( in MakeMapAndBatchNode()
77 NodeDef* tmp = graph_utils::AddScalarConstNode<int64>(1, graph); in MakeMapAndBatchNode()
85 NodeDef* tmp = graph_utils::AddScalarConstNode<bool>(false, graph); in MakeMapAndBatchNode()
Ddisable_intra_op_parallelism_test.cc44 NodeDef *start_val = graph_utils::AddScalarConstNode<int64>(0, &graph); in TEST_P()
45 NodeDef *stop_val = graph_utils::AddScalarConstNode<int64>(10, &graph); in TEST_P()
46 NodeDef *step_val = graph_utils::AddScalarConstNode<int64>(1, &graph); in TEST_P()
56 graph_utils::AddScalarConstNode<int64>(value, &graph); in TEST_P()
Dgraph_utils.cc125 NodeDef* AddScalarConstNode(bool v, MutableGraphView* graph) { in AddScalarConstNode() function
131 NodeDef* AddScalarConstNode(double v, MutableGraphView* graph) { in AddScalarConstNode() function
137 NodeDef* AddScalarConstNode(float v, MutableGraphView* graph) { in AddScalarConstNode() function
143 NodeDef* AddScalarConstNode(int v, MutableGraphView* graph) { in AddScalarConstNode() function
149 NodeDef* AddScalarConstNode(int64 v, MutableGraphView* graph) { in AddScalarConstNode() function
155 NodeDef* AddScalarConstNode(StringPiece v, MutableGraphView* graph) { in AddScalarConstNode() function
Dhoist_random_uniform.cc75 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()
Dfilter_with_random_uniform_fusion.cc60 NodeDef* tmp_rate = graph_utils::AddScalarConstNode<float>(rate, graph); in MakeFusedNode()
62 NodeDef* tmp_seed = graph_utils::AddScalarConstNode<int64>(seed, graph); in MakeFusedNode()
64 NodeDef* tmp_seed2 = graph_utils::AddScalarConstNode<int64>(seed2, graph); in MakeFusedNode()
Dmap_vectorization.cc284 auto autotune_val = graph_utils::AddScalarConstNode( in AddNewMapNode()
289 graph_utils::AddScalarConstNode(data::model::kAutotune, graph); in AddNewMapNode()
328 auto autotune_val = graph_utils::AddScalarConstNode( in AddNewPrefetchNode()
430 graph_utils::AddScalarConstNode(static_cast<int64>(1), graph); in AddNewChooseFastestNode()
Ddisable_intra_op_parallelism.cc85 graph_utils::AddScalarConstNode(int64{1}, &graph); in OptimizeAndCollectStats()
Dlatency_all_edges.cc50 NodeDef* tag = graph_utils::AddScalarConstNode<StringPiece>( in MakeLatencyNode()
Dbatch_parallelization.cc49 graph_utils::AddScalarConstNode(data::model::kAutotune, graph); in MakeParallelBatch()
Dmap_parallelization.cc47 auto* num_parallel_calls = graph_utils::AddScalarConstNode( in MakeParallelMap()
Dautotune_buffer_sizes.cc64 graph_utils::AddScalarConstNode(data::model::kAutotune, &graph); in OptimizeAndCollectStats()
Dauto_shard.cc149 graph_utils::AddScalarConstNode<int64>(num_workers, graph); in AddShardNode()
150 NodeDef* index_node = graph_utils::AddScalarConstNode<int64>(index, graph); in AddShardNode()
587 auto num_replicas_node = graph_utils::AddScalarConstNode(num_replicas, graph); in RewriteRebatchV2ToV1()
Dshuffle_and_repeat_fusion.cc86 NodeDef zero_node = *graph_utils::AddScalarConstNode<int64>(0, graph); in FuseShuffleV2AndRepeat()
Dfusion_utils.cc266 auto* node = graph_utils::AddScalarConstNode(false, &graph_view); in CreateFalsePredicate()