Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcompute_batch_size_op.cc73 const NodeDef* GetInputNode(const NodeDef& node, in GetInputNode() function
125 auto batch_size_node = GetInputNode(node, graph, arg_index); in GetBatchSizeFromBatchNode()
158 const NodeDef* input_0 = GetInputNode(node, graph, 0); in GetBatchSize()
161 const NodeDef* input = GetInputNode(node, graph, i); in GetBatchSize()
175 const NodeDef* input = GetInputNode(node, graph, 0); in GetBatchSize()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dreorder_data_discarding_ops_test.cc79 EXPECT_EQ(graph_utils::GetInputNode( in TEST()
83 EXPECT_EQ(graph_utils::GetInputNode( in TEST()
88 graph_utils::GetInputNode(*graph.GetNode("map_and_batch"), graph)->name(), in TEST()
Dreorder_data_discarding_ops.cc93 NodeDef* start_parent = graph_utils::GetInputNode(*start, graph); in OptimizeAndCollectStats()
96 start_parent = graph_utils::GetInputNode(*start, graph); in OptimizeAndCollectStats()
101 NodeDef* parent = graph_utils::GetInputNode(*discard_node, graph); in OptimizeAndCollectStats()
Dgraph_utils_test.cc278 TEST(GraphUtilsTest, GetInputNode) { in TEST() argument
285 EXPECT_EQ(GetInputNode(*node2, graph), node1); in TEST()
286 EXPECT_EQ(GetInputNode(*node1, graph), nullptr); in TEST()
297 EXPECT_EQ(GetInputNode(*node3, graph), node1); in TEST()
298 EXPECT_EQ(GetInputNode(*node3, graph, 1), node2); in TEST()
299 EXPECT_EQ(GetInputNode(*node3, graph, 0), node1); in TEST()
300 EXPECT_EQ(GetInputNode(*node3, graph, 2), nullptr); in TEST()
301 EXPECT_EQ(GetInputNode(*node1, graph), nullptr); in TEST()
Dauto_shard.cc427 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset()
431 graph_utils::GetInputNode(*input_node, *graph, 0); in FindFuncAndTensorSliceDataset()
444 const NodeDef* input_node = graph_utils::GetInputNode(*node, *graph, 0); in FindFuncAndTensorSliceDataset()
458 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp()
470 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp()
496 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp()
542 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp()
569 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in RewriteRebatchV2ToV1()
617 NodeDef* input_node = graph_utils::GetInputNode(sink_node, *graph); in ShardByData()
Dslack.cc91 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, 0); in RecursivelyHandleOp()
97 NodeDef* input_node = graph_utils::GetInputNode(*dataset_node, graph, i); in RecursivelyHandleOp()
Ddisable_intra_op_parallelism.cc71 NodeDef* last_node = graph_utils::GetInputNode(*sink_node, graph); in OptimizeAndCollectStats()
80 last_node = graph_utils::GetInputNode(*last_node, graph); in OptimizeAndCollectStats()
Dgraph_utils.h125 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph);
128 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph,
Dautotune_buffer_sizes.cc89 NodeDef* prefetched_node = graph_utils::GetInputNode(node, graph); in OptimizeAndCollectStats()
Dfilter_fusion.cc107 get_filter_node(*graph_utils::GetInputNode(*second_filter_node, graph)); in OptimizeAndCollectStats()
Dmap_vectorization.cc485 auto tmp_input_node = graph_utils::GetInputNode(*batch_node, graph); in FindMapAndBatchPattern()
488 tmp_input_node = graph_utils::GetInputNode(*tmp_input_node, graph); in FindMapAndBatchPattern()
508 input_node = graph_utils::GetInputNode(*map_node, graph); in FindMapAndBatchPattern()
Dmap_and_batch_fusion.cc124 NodeDef* node2 = graph_utils::GetInputNode(batch_node, graph); in OptimizeAndCollectStats()
Dnoop_elimination.cc149 NodeDef* const parent = graph_utils::GetInputNode(node, graph); in OptimizeAndCollectStats()
Dmap_fusion.cc128 get_map_node(*graph_utils::GetInputNode(*map_node, graph)); in OptimizeAndCollectStats()
Dgraph_utils.cc271 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph) { in GetInputNode() function
277 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph, in GetInputNode() function
Dshuffle_and_repeat_fusion.cc167 *graph_utils::GetInputNode(repeat_node, graph); in OptimizeAndCollectStats()
Dmap_and_filter_fusion.cc221 get_map_node(*graph_utils::GetInputNode(*filter_node, graph)); in OptimizeAndCollectStats()
Dhoist_random_uniform.cc270 const NodeDef& parent_node = *graph_utils::GetInputNode(*map_node, graph); in OptimizeAndCollectStats()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc361 TF_RETURN_IF_ERROR(GetInputNode(*input_tensor, &input_node)); in AbsorbInputByOptimizedNodesGroup()
734 TF_RETURN_IF_ERROR(GetInputNode(node->input(0), &input_0)); in TrySimplify()
809 TF_RETURN_IF_ERROR(GetInputNode(node->input(i), &input)); in GetCommonFactors()
878 TF_RETURN_IF_ERROR(GetInputNode(input, &inner_node)); in GetUniqueFactors()
1129 TF_RETURN_IF_ERROR(GetInputNode(tail->input(0), &first_transpose)); in TrySimplify()
1132 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &node_perm)); in TrySimplify()
1142 GetInputNode(first_transpose->input(1), &first_transpose_perm)); in TrySimplify()
1247 TF_RETURN_IF_ERROR(GetInputNode(tail->input(0), &involution)); in TrySimplify()
1296 TF_RETURN_IF_ERROR(GetInputNode(node->name(), &bitcast)); in TrySimplify()
1298 TF_RETURN_IF_ERROR(GetInputNode(node->input(0), &operand)); in TrySimplify()
[all …]
Dgraph_optimizer_stage_test.cc165 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
207 TF_CHECK_OK(stage.GetInputNode("Add", &add_node)); in TEST_F()
219 TF_CHECK_OK(stage.GetInputNode("Add_1", &add_node_copy_by_name)); in TEST_F()
229 TF_CHECK_OK(stage.GetInputNode("Add_2", &empty_node_by_name)); in TEST_F()
Dgraph_optimizer_stage.h69 Status GetInputNode(const GraphOptimizerContext& ctx, const string& input,
186 Status GetInputNode(const string& input, NodeDef** node) const { in GetInputNode() function
187 return ::tensorflow::grappler::GetInputNode(ctx_, input, node); in GetInputNode()
Dgraph_optimizer_stage.cc31 Status GetInputNode(const GraphOptimizerContext& ctx, const string& input, in GetInputNode() function