- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for mutable_scoped_allocator_opts (0.11 sec)
-
tensorflow/c/eager/c_api_test_util.cc
auto* rewrite_options = config->mutable_graph_options()->mutable_rewrite_options(); rewrite_options->set_scoped_allocator_optimization( tensorflow::RewriterConfig::ON); rewrite_options->mutable_scoped_allocator_opts()->add_enable_op( "CollectiveReduce"); if ((tensorflow::IsGoogleCudaEnabled() || tensorflow::IsBuiltWithROCm()) && num_virtual_gpus > 0) { tensorflow::GPUOptions* gpu_options =
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K bytes - Viewed (0)