Home
last modified time | relevance | path

Searched refs:mutable_graph_options (Results 1 – 25 of 49) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer_test.cc142 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
161 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
183 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
205 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
228 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
245 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
264 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
520 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
633 *config_proto.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
736 *config.mutable_graph_options()->mutable_rewrite_options(); in TEST_F()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/
H A Dcluster.cc41 options_.config.mutable_graph_options()->set_build_cost_model_after( in SetNumWarmupSteps()
56 options_.config.mutable_graph_options()->set_build_cost_model(0); in DisableDetailedStats()
59 options_.config.mutable_graph_options()->set_build_cost_model(1); in DisableDetailedStats()
70 options_.config.mutable_graph_options()->mutable_optimizer_options(); in DisableOptimizer()
75 options_.config.mutable_graph_options()->mutable_rewrite_options(); in DisableOptimizer()
92 options_.config.mutable_graph_options()->mutable_rewrite_options(); in DisableOptimizer()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Ddirect_session_with_tracking_alloc_test.cc73 options.config.mutable_graph_options()->set_build_cost_model(1); in TEST()
74 options.config.mutable_graph_options() in TEST()
77 options.config.mutable_graph_options() in TEST()
80 options.config.mutable_graph_options() in TEST()
145 options.config.mutable_graph_options()->set_build_cost_model(1); in TEST()
146 options.config.mutable_graph_options()->set_build_cost_model_after( in TEST()
194 options.config.mutable_graph_options()->set_build_cost_model(1); in TestHWAccelerator()
275 options.config.mutable_graph_options()->set_build_cost_model(1); in TEST()
276 options.config.mutable_graph_options() in TEST()
335 options.config.mutable_graph_options() in TEST()
H A Dlower_function_call_op_test.cc53 session_options.config.mutable_graph_options() in SessionOptionsWithInlining()
H A Dlower_functional_ops_test.cc50 session_options.config.mutable_graph_options() in SessionOptionsWithInlining()
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dtf_wrapper_test.cc35 config_proto.mutable_graph_options()->set_timeline_step(123); in TEST()
49 config_proto.mutable_graph_options()->set_timeline_step(123); in TEST()
75 config_proto.mutable_graph_options()->set_timeline_step(123); in TEST()
98 expected_config_proto.mutable_graph_options()->set_place_pruned_graph(true); in TEST()
118 expected_config_proto.mutable_graph_options()->set_place_pruned_graph(true); in TEST()
H A Dtf_wrapper.cc76 config_proto.mutable_graph_options()->set_place_pruned_graph(true); in InitializeConfigProto()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_execution_options.cc29 config.mutable_graph_options() in CreateDefaultSessionOptions()
40 config.mutable_graph_options() in CreateDefaultSessionOptions()
44 config.mutable_graph_options() in CreateDefaultSessionOptions()
49 config.mutable_graph_options() in CreateDefaultSessionOptions()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtransform_graph_function.cc119 session_options.config.mutable_graph_options() in TransformGraphFunction()
148 config_proto.mutable_graph_options() in TransformGraphFunction()
152 config_proto.mutable_graph_options() in TransformGraphFunction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc161 config_proto.mutable_graph_options()->mutable_rewrite_options()->CopyFrom( in RunTfTrt()
378 opts.config.mutable_graph_options()->mutable_rewrite_options(); in GetSessionConfg()
433 config_proto.mutable_graph_options()->mutable_rewrite_options(); in InlineFunctions()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dimmutable_constant_op_test.cc107 session_options.config.mutable_graph_options() in TEST()
180 session_options.config.mutable_graph_options() in TEST()
H A Dmatmul_op_test.cc58 session_options.config.mutable_graph_options() in RunAndFetch()
64 session_options.config.mutable_graph_options() in RunAndFetch()
H A Dfused_batch_norm_ex_op_test.cc97 session_options.config.mutable_graph_options()
103 session_options.config.mutable_graph_options()
H A Dpartitioned_function_ops.cc63 config_proto_->mutable_graph_options() in PartitionedCallOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgrappler_test.cc101 options_.config.mutable_graph_options()->mutable_rewrite_options(), in DisableAllOptimizers()
107 options_.config.mutable_graph_options()->mutable_rewrite_options(), in EnableAllOptimizers()
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Dgrpc_session_debug_test.cc86 options.config.mutable_graph_options() in Options()
89 options.config.mutable_graph_options() in Options()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass_test_helper.cc37 session_options.config.mutable_graph_options() in MarkForCompilation()
H A Dxla_activity_listener_test.cc95 options.config.mutable_graph_options() in TEST_F()
H A Dcluster_scoping_pass_test.cc38 wrapper.session_options.config.mutable_graph_options() in ClusterScoping()
H A Dclone_constants_for_better_clustering_test.cc35 session_options.config.mutable_graph_options() in CloneConstantsForBetterClustering()
/aosp_15_r20/external/tensorflow/tensorflow/tools/optimization/
H A Doptimization_pass_runner.cc97 session_options->config.mutable_graph_options() in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dxla_compilation_cache_test_helper.cc111 *options.config.mutable_graph_options()->mutable_optimizer_options(); in ExecuteWithBatch()
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc70 config.mutable_graph_options()->mutable_optimizer_options(); in TF_EnableXLACompilation()
134 config.mutable_graph_options()->mutable_optimizer_options(); in TF_CreateConfig()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model_testutil.cc107 session_options.config.mutable_graph_options() in ComputeCurrentTFResult()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.cc123 *config.mutable_graph_options()->mutable_rewrite_options() = config_factory(); in ApplyRewrites()

12