/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
H A D | fusion_utils_test.cc | 56 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 D | map_and_filter_fusion.cc | 209 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 D | function_utils_test.cc | 801 FunctionDef* function = graph_def.mutable_library()->add_function(); in TEST() 805 *graph_def.mutable_library()); in TEST()
|
H A D | filter_fusion.cc | 101 output->mutable_library()); in OptimizeAndCollectStats()
|
H A D | map_fusion.cc | 121 fusion_utils::MergeNodes, output->mutable_library()); in OptimizeAndCollectStats()
|
H A D | meta_optimizer.cc | 157 *output->mutable_library() = flib.ToProto(); in Optimize()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
H A D | graph_def_util_test.cc | 203 *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 D | hash_utils_test.cc | 41 *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 D | rewrite_utils.cc | 131 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 D | tpu_test.cc | 33 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 D | batch_op_rewriter.cc | 166 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 D | meta_optimizer.cc | 921 *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 D | tfg_optimizer_hook.cc | 158 (void)graphdef.mutable_library(); in Optimize()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
H A D | cluster_function_library_runtime.cc | 41 *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 D | graphdef_export.cc | 200 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 D | tfrt_graph_execution_state.cc | 177 *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 D | tf_mlir_translate.cc | 96 pruned_graph_def.mutable_library()->Swap(graphdef.mutable_library()); in GraphdefToMlirImport()
|
H A D | upgrade_graph.cc | 78 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/ |
D | graph.pb.h | 198 ::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 D | tf2xla.cc | 149 for (auto& fn : *graph_def->mutable_library()->mutable_function()) { in ConvertVarHandlesToAotVarHandles()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
H A D | resolve_cluster.cc | 150 *(pruned_graph->mutable_library()) = tf_graph.library(); in MaybeReplaceCompositeSubgraph()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/util/tensor_bundle/ |
H A D | byte_swap.cc | 150 ->mutable_library() in ByteSwapTensorContent()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/ |
H A D | roundtrip.cc | 116 FunctionDefLibrary* library = graphdef.mutable_library(); in NormalizeTensorData()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/function/ |
H A D | runtime_client.cc | 132 *graph.mutable_library()->add_function() = *fn; in TransformFunction()
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/saved_model/ |
H A D | fingerprinting.cc | 151 graph_def.mutable_library()->Clear(); in CanonicalizeGraphDef()
|