- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for set_output_partition_graphs (0.24 sec)
-
tensorflow/cc/framework/while_gradients_test.cc
// Run with two CPU devices and output partition graphs SessionOptions session_options; (*session_options.config.mutable_device_count())["CPU"] = 2; RunOptions run_options; run_options.set_output_partition_graphs(true); RunMetadata run_metadata; Run<int>(ClientSession(scope_, session_options), {0, 1}, {1, 11}, run_options, &run_metadata); // Check that at least one node ran on each device
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 27 20:32:17 UTC 2017 - 7.7K bytes - Viewed (0) -
tensorflow/cc/saved_model/saved_model_bundle_lite_test.cc
outputs[0], test::AsTensor<float>({2, 2.5, 3, 3.5}, TensorShape({4, 1}))); // Validate the `output_partition_graphs` is not supported. RunOptions run_options; run_options.set_output_partition_graphs(true); RunMetadata run_metadata; Status s = bundle.GetSession()->Run(run_options, {{input_name, input}}, {output_name}, {}, &outputs, &run_metadata);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 19:16:58 UTC 2023 - 9.7K bytes - Viewed (0)