| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | graph_utils_test.cc | 220 TEST(GraphUtilsTest, GetInputNode) { in TEST() argument
227 EXPECT_EQ(GetInputNode(*node2, graph), node1); in TEST()
228 EXPECT_EQ(GetInputNode(*node1, graph), nullptr); in TEST()
239 EXPECT_EQ(GetInputNode(*node3, graph), node1); in TEST()
240 EXPECT_EQ(GetInputNode(*node3, graph, 1), node2); in TEST()
241 EXPECT_EQ(GetInputNode(*node3, graph, 0), node1); in TEST()
242 EXPECT_EQ(GetInputNode(*node3, graph, 2), nullptr); in TEST()
243 EXPECT_EQ(GetInputNode(*node1, graph), nullptr); in TEST()
|
| D | rebatch.cc | 145 NodeDef* batch_size_node = graph_utils::GetInputNode(node, *graph, 1); in MutateBatchSize()
194 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp()
201 NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp()
|
| D | graph_utils.h | 109 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph);
112 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph,
|
| D | auto_shard.cc | 187 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, i); in RecursivelyHandleOp()
222 const NodeDef* input_node = graph_utils::GetInputNode(node, *graph, 0); in RecursivelyHandleOp()
|
| D | shuffle_and_repeat_fusion.cc | 79 NodeDef* node2 = graph_utils::GetInputNode(repeat_node, graph); in OptimizeAndCollectStats()
|
| D | noop_elimination.cc | 84 NodeDef* const parent = graph_utils::GetInputNode(node, graph); in OptimizeAndCollectStats()
|
| D | map_vectorization.cc | 464 auto tmp_input_node = graph_utils::GetInputNode(*batch_node, graph); in FindMapAndBatchPattern()
467 tmp_input_node = graph_utils::GetInputNode(*tmp_input_node, graph); in FindMapAndBatchPattern()
486 input_node = graph_utils::GetInputNode(*map_node, graph); in FindMapAndBatchPattern()
|
| D | map_and_batch_fusion.cc | 116 NodeDef* node2 = graph_utils::GetInputNode(batch_node, graph); in OptimizeAndCollectStats()
|
| D | filter_fusion.cc | 106 get_filter_node(*graph_utils::GetInputNode(*second_filter_node, graph)); in OptimizeAndCollectStats()
|
| D | graph_utils.cc | 229 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph) { in GetInputNode() function
235 NodeDef* GetInputNode(const NodeDef& node, const MutableGraphView& graph, in GetInputNode() function
|
| D | map_fusion.cc | 128 get_map_node(*graph_utils::GetInputNode(*map_node, graph)); in OptimizeAndCollectStats()
|
| D | map_and_filter_fusion.cc | 149 get_map_node(*graph_utils::GetInputNode(*filter_node, graph)); in OptimizeAndCollectStats()
|
| D | hoist_random_uniform.cc | 260 const NodeDef& parent_node = *graph_utils::GetInputNode(*map_node, graph); in OptimizeAndCollectStats()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 336 TF_RETURN_IF_ERROR(GetInputNode(*input_tensor, &input_node)); in AbsorbInputByOptimizedNodesGroup()
708 TF_RETURN_IF_ERROR(GetInputNode(node->input(0), &input_0)); in TrySimplify()
783 TF_RETURN_IF_ERROR(GetInputNode(node->input(i), &input)); in GetCommonFactors()
851 TF_RETURN_IF_ERROR(GetInputNode(input, &inner_node)); in GetUniqueFactors()
1101 TF_RETURN_IF_ERROR(GetInputNode(tail->input(0), &first_transpose)); in TrySimplify()
1104 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &node_perm)); in TrySimplify()
1114 GetInputNode(first_transpose->input(1), &first_transpose_perm)); in TrySimplify()
1207 TF_RETURN_IF_ERROR(GetInputNode(tail->input(0), &involution)); in TrySimplify()
1256 TF_RETURN_IF_ERROR(GetInputNode(node->name(), &bitcast)); in TrySimplify()
1258 TF_RETURN_IF_ERROR(GetInputNode(node->input(0), &operand)); in TrySimplify()
[all …]
|
| D | graph_optimizer_stage_test.cc | 165 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()
|
| D | graph_optimizer_stage.h | 69 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()
|
| D | graph_optimizer_stage.cc | 31 Status GetInputNode(const GraphOptimizerContext& ctx, const string& input, in GetInputNode() function
|