Home
last modified time | relevance | path

Searched refs:mutable_node (Results 1 – 25 of 124) sorted by relevance

12345

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsparsify_gather.cc216 g->mutable_node()->SwapElements(i, i + 1); in RemoveNodeAtIndex()
218 g->mutable_node()->RemoveLast(); in RemoveNodeAtIndex()
455 init_op = replaced_graph_def.mutable_node(i); in SparsifyGatherInternal()
461 init_op = replaced_graph_def.mutable_node()->Add(); in SparsifyGatherInternal()
512 RemoveInputAtIndex(replaced_graph_def.mutable_node(i), j)); in SparsifyGatherInternal()
522 replaced_graph_def.mutable_node(i)); in SparsifyGatherInternal()
531 replaced_graph_def.mutable_node(i)->mutable_input()->RemoveLast(); in SparsifyGatherInternal()
532 replaced_graph_def.mutable_node(i)->mutable_attr()->erase("N"); in SparsifyGatherInternal()
533 replaced_graph_def.mutable_node(i)->set_op("Identity"); in SparsifyGatherInternal()
H A Dtransform_utils.cc163 output_graph_def->mutable_node()->Clear(); in FilterGraphDef()
166 *output_graph_def->mutable_node()->Add() = node; in FilterGraphDef()
174 output_graph_def->mutable_node()->Clear(); in RemoveAttributes()
176 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RemoveAttributes()
240 *output_graph_def->mutable_node()->Add() = node_def; in SortByExecutionOrder()
455 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes()
460 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes()
466 NodeDef* added_node = output_graph_def->mutable_node()->Add(); in ReplaceMatchingOpTypes()
490 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in RenameNodeInputs()
H A Dtransform_utils_test.cc705 NodeDef* mul_node = graph_def.mutable_node()->Add(); in TestFindInvalidInputs()
712 NodeDef* add_node1 = graph_def.mutable_node()->Add(); in TestFindInvalidInputs()
719 NodeDef* add_node2 = graph_def.mutable_node()->Add(); in TestFindInvalidInputs()
726 NodeDef* const_node1 = graph_def.mutable_node()->Add(); in TestFindInvalidInputs()
730 NodeDef* const_node2 = graph_def.mutable_node()->Add(); in TestFindInvalidInputs()
753 NodeDef* mul_node = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid()
760 NodeDef* add_node1 = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid()
767 NodeDef* add_node2 = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid()
774 NodeDef* const_node1 = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid()
778 NodeDef* const_node2 = invalid_graph_def.mutable_node()->Add(); in TestIsGraphValid()
[all …]
H A Dfold_old_batch_norms_test.cc100 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldOldBatchNorms()
178 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
256 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldFusedBatchNorms()
333 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
434 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldFusedBatchNormsWithConcat()
526 *(original_graph_def.mutable_node()->Add()) = batch_norm_node; in TestFoldFusedBatchNormsWithBatchToSpace()
H A Dinline_partitionedcall.cc111 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in InlinePartitionedCall()
128 NodeDef* new_node = output_graph_def->mutable_node()->Add(); in InlinePartitionedCall()
134 for (NodeDef& node : *output_graph_def->mutable_node()) { in InlinePartitionedCall()
H A Dfreeze_requantization_ranges.cc179 NodeDef* min_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
189 NodeDef* max_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
198 NodeDef* new_node = frozen_graph_def.mutable_node()->Add(); in FreezeRequantizationRanges()
H A Dquantize_nodes.cc236 NodeDef* new_node = merged_graph_def.mutable_node()->Add(); in MergeDuplicateNodes()
330 *(placeholder_graph_def.mutable_node()->Add()) = node; in QuantizePlaceholders()
337 *(placeholder_graph_def.mutable_node()->Add()) = quantized_placeholder; in QuantizePlaceholders()
346 *(placeholder_graph_def.mutable_node()->Add()) = min_node; in QuantizePlaceholders()
355 *(placeholder_graph_def.mutable_node()->Add()) = max_node; in QuantizePlaceholders()
366 *(placeholder_graph_def.mutable_node()->Add()) = dequantize_node; in QuantizePlaceholders()
H A Dinsert_logging.cc94 NodeDef* new_node = logged_graph_def.mutable_node()->Add(); in InsertLogging()
114 NodeDef* print_node = logged_graph_def.mutable_node()->Add(); in InsertLogging()
H A Dstrip_unused_nodes.cc183 *(output_graph_def->mutable_node()->Add()) = placeholder_node; in StripUnusedNodes()
185 *(output_graph_def->mutable_node()->Add()) = node; in StripUnusedNodes()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper_test.cc180 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
285 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
416 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
556 item.graph.mutable_node(i)->set_device("/device:GPU:0"); in TEST_F()
680 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
825 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
950 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
1047 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
1113 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
1179 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
[all …]
H A Dmodel_pruner.cc433 pruned_graph->mutable_node()->Reserve(item.graph.node_size()); in Optimize()
463 NodeDef* node = pruned_graph->mutable_node(i); in Optimize()
506 optimized_graph->mutable_node()->Swap(pruned_graph->mutable_node()); in Optimize()
512 optimized_graph->mutable_node()->Reserve(pruned_graph->node_size()); in Optimize()
H A Dmkl_remapper_test.cc123 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in FuseConv2DWithBiasAndAddNOrAdd()
266 item.graph.mutable_node(i)->set_device("/device:CPU:0"); \
379 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
499 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in TEST_F()
592 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RelpaceAddWithBiasAddDepthwiseConv2D()
698 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
798 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in VerifyFused()
877 item.graph.mutable_node(i)->set_device("/device:CPU:0"); in RunTest()
H A Dshape_optimizer.cc75 for (auto& node : *optimized_graph->mutable_node()) { in Optimize()
143 for (auto& node : *optimized_graph->mutable_node()) { in Optimize()
H A Dtfg_optimizer_hook_test.cc113 item.graph.mutable_node(0)->mutable_attr()->insert({"", std::move(attr)}); in TEST()
155 item.graph.mutable_node(0)->mutable_attr()->insert({"", std::move(attr)}); in TEST()
H A Ddependency_optimizer.cc257 NodeDef* node = optimized_graph_->mutable_node(node_idx); in OptimizeNode()
470 DedupControlInputs(optimized_graph_->mutable_node(i)); in CleanControlInputs()
613 NodeDef* target_node = optimized_graph_->mutable_node(target); in TransitiveReduction()
653 NodeDef* node = optimized_graph_->mutable_node(i); in GroupCrossDeviceControlEdges()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dgraph.pb.h178 ::tensorflow::NodeDef* mutable_node(int index); in Swap()
180 mutable_node(); in Swap()
270 inline ::tensorflow::NodeDef* GraphDef::mutable_node(int index) { in mutable_node() function
275 GraphDef::mutable_node() { in mutable_node() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dcanonicalizer.cc51 CanonicalizeNode(graph->mutable_node(i)); in CanonicalizeGraph()
57 NodeDef* node = graph->mutable_node(i); in CompressConstants()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dmake_deterministic.cc165 return graph->graph()->mutable_node(index); in GetMutableNode()
682 NodeDef* mutable_node = GetMutableNode(node.name(), &graph); in OptimizeAndCollectStats() local
683 (*mutable_node->mutable_attr())["sloppy"].set_b(false); in OptimizeAndCollectStats()
687 NodeDef* mutable_node = GetMutableNode(node.name(), &graph); in OptimizeAndCollectStats() local
688 (*mutable_node->mutable_attr())["deterministic"].set_s("true"); in OptimizeAndCollectStats()
/aosp_15_r20/external/sandboxed-api/sandboxed_api/sandbox2/
H A Dmounts.h43 *mount_tree_.mutable_node() = root; in Mounts()
79 mount_tree_.mutable_node()->mutable_root_node()->set_writable(true); in SetRootWritable()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Dmutable_graph_view.h46 for (NodeDef& node : *graph->mutable_node()) AddUniqueNodeOrDie(&node); in MutableGraphView()
47 for (NodeDef& node : *graph->mutable_node()) AddAndDedupFanouts(&node); in MutableGraphView()
H A Dutils.cc343 graph->mutable_node()->SwapElements(n, r); in PermuteNodesInPlace()
374 graph->mutable_node()->SwapElements(index, last); in EraseNodesFromGraphImpl()
377 graph->mutable_node()->DeleteSubrange(last + 1, nodes_to_delete.size()); in EraseNodesFromGraphImpl()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Droundtrip.cc123 NormalizeNode(graphdef.mutable_node(i), add_fulltype); in NormalizeTensorData()
124 llvm::sort(*graphdef.mutable_node(), in NormalizeTensorData()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
H A Dbatch_op_rewriter.cc159 NodeDef* node = optimized_graph->mutable_node(i); in Optimize()
188 NodeDef* node = optimized_graph->mutable_node(i); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state.cc552 for (auto& node : *graph_def.mutable_node()) { in PruneGraphDef()
693 for (auto& node : *graph_def.mutable_node()) { in EliminateRefVariablesFromV1ControlFlow()
751 graph_def.mutable_node()->Swap(updated_graph_def.mutable_node()); in EliminateRefVariablesFromV1ControlFlow()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_def_util.cc72 NodeDef* node_def = graph_def->mutable_node(i); in AddDefaultAttrsToGraphDef()
144 NodeDef* node_def = graph_def->mutable_node(n); in RemoveNewDefaultAttrsFromGraphDef()

12345