Home
last modified time | relevance | path

Searched refs:mutable_library (Results 1 – 25 of 62) sorted by relevance

123

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dfusion_utils_test.cc56 auto *parent_function = graph.mutable_library()->add_function(); in TEST()
58 auto *function = graph.mutable_library()->add_function(); in TEST()
64 fusion_utils::MergeNodes, graph.mutable_library()); in TEST()
94 auto *xtimes_two = graph.mutable_library()->add_function(); in TEST()
96 auto *is_zero = graph.mutable_library()->add_function(); in TEST()
103 graph.mutable_library()); in TEST()
134 auto *parent_function = graph.mutable_library()->add_function(); in TEST()
136 auto *function = graph.mutable_library()->add_function(); in TEST()
142 fusion_utils::MergeNodes, graph.mutable_library()); in TEST()
152 auto *function = graph.mutable_library()->add_function(); in TEST()
[all …]
H A Dmap_and_filter_fusion.cc209 output->mutable_library()); in OptimizeAndCollectStats()
227 *fused_maps, *fused_function, &graph, output->mutable_library())); in OptimizeAndCollectStats()
231 &graph, output->mutable_library())); in OptimizeAndCollectStats()
H A Dfunction_utils_test.cc801 FunctionDef* function = graph_def.mutable_library()->add_function(); in TEST()
805 *graph_def.mutable_library()); in TEST()
H A Dfilter_fusion.cc101 output->mutable_library()); in OptimizeAndCollectStats()
H A Dmap_fusion.cc121 fusion_utils::MergeNodes, output->mutable_library()); in OptimizeAndCollectStats()
H A Dmeta_optimizer.cc157 *output->mutable_library() = flib.ToProto(); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dgraph_def_util_test.cc203 *produced_graph_def.mutable_library()->add_function() = in TEST()
222 *expected_graph_def.mutable_library()->add_function() = in TEST()
291 FunctionDef* function_def = graph_def.mutable_library()->add_function(); in TEST()
337 FunctionDef* b = graph_def.mutable_library()->add_function(); in TEST()
338 FunctionDef* c = graph_def.mutable_library()->add_function(); in TEST()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc41 *graph_def.mutable_library() = library; in GetHash()
56 *graph_def.mutable_library() = library; in CheckEqual()
481 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
545 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
612 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
660 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
712 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
783 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
1000 FunctionDefLibrary* fl1 = gd.mutable_library(); in TEST_F()
1182 FunctionDefLibrary* fl = graph_def.mutable_library(); in BM_ParallelFunctionCallsGraph()
[all …]
H A Drewrite_utils.cc131 for (auto& function_def : *graph_def->mutable_library()->mutable_function()) { in ApplyRewrites()
272 *graph_def->mutable_library()->mutable_function()) { in GetGrapplerItem()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dtpu_test.cc33 FunctionDefLibrary* library = tpu_graph.mutable_library(); in TEST_F()
54 FunctionDefLibrary* library = tpu_graph.mutable_library(); in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
H A Dbatch_op_rewriter.cc166 optimized_graph->mutable_library()->mutable_function(i); in Optimize()
197 optimized_graph->mutable_library()->mutable_function(i); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer.cc921 *optimized_graph->mutable_library() = in RunOptimizer()
1063 *item.graph.mutable_library() = minimized_flib(item.graph).ToProto(); in OptimizeConsumeItem()
1150 PropagateTFDataAttrs(flib, *optimized_graph->mutable_library()); in OptimizeConsumeItem()
1237 *func_item.graph.mutable_library() = in OptimizeConsumeItem()
1268 *optimized_graph->mutable_library() = flib.ToProto(); in OptimizeConsumeItem()
1399 *item.graph.mutable_library() = flib->ToProto(); in OptimizeGraph()
H A Dtfg_optimizer_hook.cc158 (void)graphdef.mutable_library(); in Optimize()
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dcluster_function_library_runtime.cc41 *register_function->mutable_library()->mutable_function()) { in StripDefaultAttributesInRegisterFunctionOp()
93 *register_function->mutable_library() = in Instantiate()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/
H A Dgraphdef_export.cc200 args.push_back(Argument{func, graph->mutable_library()->add_function()}); in ExportToGraphDef()
218 func, graph->mutable_library()->add_function(), gradient)); in ExportToGraphDef()
220 *graph->mutable_library()->add_gradient() = std::move(*gradient); in ExportToGraphDef()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state.cc177 *graph_def.mutable_library() = flib_def.ToProto(); in CreateGraphDefFromGraphAndFlibDef()
757 *graph_def.mutable_library()->mutable_function()) { in RemoveInputShapesInFunctions()
787 *graph_def.mutable_library() = flib.ToProto(); in OptimizeFunctions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dtf_mlir_translate.cc96 pruned_graph_def.mutable_library()->Swap(graphdef.mutable_library()); in GraphdefToMlirImport()
H A Dupgrade_graph.cc78 FunctionDefLibrary* library = gdef.mutable_library(); in GenerateResourceSharedNameIfEmpty()
/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.h198 ::tensorflow::FunctionDefLibrary* mutable_library(); in Swap()
454 inline ::tensorflow::FunctionDefLibrary* GraphDef::mutable_library() { in mutable_library() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dtf2xla.cc149 for (auto& fn : *graph_def->mutable_library()->mutable_function()) { in ConvertVarHandlesToAotVarHandles()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_cluster.cc150 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/
H A Dbyte_swap.cc150 ->mutable_library() in ByteSwapTensorContent()
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
H A Droundtrip.cc116 FunctionDefLibrary* library = graphdef.mutable_library(); in NormalizeTensorData()
/aosp_15_r20/external/tensorflow/tensorflow/core/function/
H A Druntime_client.cc132 *graph.mutable_library()->add_function() = *fn; in TransformFunction()
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/
H A Dfingerprinting.cc151 graph_def.mutable_library()->Clear(); in CanonicalizeGraphDef()

123