| /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()
|