Home
last modified time | relevance | path

Searched refs:mutable_parameter_map (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/data/service/
H A Dauto_shard_rewriter.cc133 (*config.mutable_parameter_map())[AutoShardDatasetOp::kNumWorkers].set_i( in GetRewriteConfig()
135 (*config.mutable_parameter_map())[AutoShardDatasetOp::kIndex].set_i( in GetRewriteConfig()
137 (*config.mutable_parameter_map())[AutoShardDatasetOp::kAutoShardPolicy].set_i( in GetRewriteConfig()
141 (*config.mutable_parameter_map())[AutoShardDatasetOp::kNumReplicas].set_i(1); in GetRewriteConfig()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/
H A Dslack_test.cc70 (*config.mutable_parameter_map())["slack_period"].set_s( in TEST_P()
103 (*config.mutable_parameter_map())["slack_period"].set_s("0"); in TEST()
164 (*config.mutable_parameter_map())["slack_period"].set_s("8"); in TEST()
H A Denable_gradient_descent_test.cc36 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithEnableGradientDescent()
38 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithEnableGradientDescent()
H A Ddisable_prefetch_legacy_autotune_test.cc39 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithDisablePrefetchLegacyAutotune()
41 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithDisablePrefetchLegacyAutotune()
H A Dfilter_parallelization_test.cc36 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithFilterParallelization()
38 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithFilterParallelization()
H A Dmap_parallelization_test.cc36 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithMapParallelization()
38 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithMapParallelization()
H A Dbatch_parallelization_test.cc36 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithBatchParallelization()
38 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithBatchParallelization()
H A Dinject_prefetch_test.cc41 (*config.mutable_parameter_map())["autotune"].set_s("true"); in Optimize()
43 (*config.mutable_parameter_map())["autotune"].set_s("false"); in Optimize()
H A Dautotune_buffer_sizes_test.cc36 (*config.mutable_parameter_map())["autotune"].set_s("true"); in OptimizeWithAutotuneBufferSizes()
38 (*config.mutable_parameter_map())["autotune"].set_s("false"); in OptimizeWithAutotuneBufferSizes()
H A Dmeta_optimizer.cc90 (*optimizer_config->mutable_parameter_map())[config_key].set_s( in ToConfigMap()
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Drewrite_utils.cc149 (*custom_optimizer->mutable_parameter_map())[kOptimizers].mutable_list(); in CreateRewriterConfig()
162 (*custom_optimizer->mutable_parameter_map())[kOptimizerConfigs] in CreateRewriterConfig()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/
H A Dauto_shard_dataset_op.cc101 (*custom_optimizer->mutable_parameter_map())[pair.first] = attr; in CreateConfig()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/inference/
H A Dbatch_op_rewriter_test.cc84 (*rewriter_config.mutable_parameter_map())["batch_op_rewrite_config"].set_s( in MakeConfig()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/
H A Dtrt_convert_api.cc133 auto trt_parameter_map = trt_optimizer->mutable_parameter_map(); in GetTrtRewriterConfig()
/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/protobuf/
Drewriter_config.pb.h618 mutable_parameter_map(); in Swap()
1567 RewriterConfig_CustomGraphOptimizer::mutable_parameter_map() { in mutable_parameter_map() function
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmeta_optimizer_test.cc165 (*custom_config->mutable_parameter_map())["foo"] = AttrValue(); in TEST_F()