| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 240 NodeDef* added_node = node_map->GetNode(ctrl_dep_name); in AddControlDependency() local
241 if (added_node == nullptr) { in AddControlDependency()
242 added_node = graph->add_node(); in AddControlDependency()
243 added_node->set_name(ctrl_dep_name); in AddControlDependency()
244 added_node->set_op("Identity"); in AddControlDependency()
245 added_node->set_device(node.device()); in AddControlDependency()
247 (*added_node->mutable_attr())["T"].set_type(output_type); in AddControlDependency()
248 *added_node->add_input() = input_name; in AddControlDependency()
249 node_map->AddNode(added_node->name(), added_node); in AddControlDependency()
250 node_map->AddOutput(node.name(), added_node->name()); in AddControlDependency()
[all …]
|
| D | layout_optimizer.cc | 630 NodeDef* added_node = graph_->add_node(); in UpdateAttrValueOfInput() local
631 *added_node = *input_node; in UpdateAttrValueOfInput()
634 added_node->set_name(node_name); in UpdateAttrValueOfInput()
636 node_map_->AddNode(node_name, added_node); in UpdateAttrValueOfInput()
638 return UpdateAttrValue(added_node, permute); in UpdateAttrValueOfInput()
1016 NodeDef* added_node = graph_->add_node(); in AddNodeDataFormatOp() local
1017 added_node->set_name(name); in AddNodeDataFormatOp()
1018 added_node->set_op(op); in AddNodeDataFormatOp()
1019 node_map_->AddNode(added_node->name(), added_node); in AddNodeDataFormatOp()
1020 added_node->set_device(node_->device()); in AddNodeDataFormatOp()
[all …]
|
| D | generic_layout_optimizer_transposer.h | 115 utils::MutationNewNode* added_node);
125 absl::string_view control_node_name, utils::MutationNewNode* added_node,
149 utils::MutationNewNode* added_node);
|
| D | generic_layout_optimizer_transposer.cc | 293 utils::MutationNewNode* added_node) { in CreateConstPermNode() argument
319 *added_node = in CreateConstPermNode()
328 absl::string_view control_node_name, utils::MutationNewNode* added_node, in CreateTransposeNode() argument
370 *added_node = in CreateTransposeNode()
445 utils::MutationNewNode* added_node) { in CreateDataFormatNode() argument
481 *added_node = in CreateDataFormatNode()
509 utils::MutationNewNode added_node; in UpdateEdge() local
528 control_node_name, &added_node, &added_node_name)); in UpdateEdge()
539 is_src_format_to_dst_format, &added_node)); in UpdateEdge()
550 mutation->AddOrUpdateRegularFanin(added_node, 0, in UpdateEdge()
|
| D | generic_layout_optimizer_transposer_test.cc | 520 utils::MutationNewNode added_node; in TEST_F() local
523 {0, 3, 1, 2}, "", &added_node)); in TEST_F()
567 utils::MutationNewNode added_node; in TEST_F() local
571 "", &added_node, &transpose_node_name)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | disable_intra_op_parallelism.cc | 108 auto* added_node = graph.AddNode(std::move(insert_node)); in OptimizeAndCollectStats() local
110 graph.UpdateFanouts(last_node->name(), added_node->name())); in OptimizeAndCollectStats()
|
| D | autotune_buffer_sizes.cc | 130 auto* added_node = graph.AddNode(std::move(prefetch_node)); in OptimizeAndCollectStats() local
132 graph.UpdateFanouts(async_dataset_node->name(), added_node->name())); in OptimizeAndCollectStats()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.cc | 455 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
456 *added_node = old_node; in ReplaceMatchingOpTypes()
460 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
461 *added_node = new_node; in ReplaceMatchingOpTypes()
466 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes() local
467 *added_node = input_node; in ReplaceMatchingOpTypes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | inline_function_utils.cc | 617 Status added_node; in InlineFunctionBody() local
618 Node* clone = g->AddNode(ndef, &added_node); in InlineFunctionBody()
619 TF_CHECK_OK(added_node); in InlineFunctionBody()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 585 NodeDef* added_node = graph_def.add_node(); in TEST() local
586 *added_node = node_def; in TEST()
|