- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getDeviceNames (0.11 sec)
-
tensorflow/c/experimental/grappler/grappler_test.cc
TF_ASSERT_OK(cluster->Provision()); TrivialTestGraphInputYielder fake_input(4, 1, 10, false, cluster->GetDeviceNames()); GrapplerItem item; CHECK(fake_input.NextItem(&item)); TF_Status* status = TF_NewStatus(); TF_GraphProperties* graph_properties =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 13 22:30:58 UTC 2023 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
// tf.Case. This could be enhanced to have some sort of policy in the // future. OpBuilder b(op); RankedTensorType type = RankedTensorType::get({}, b.getIntegerType(32)); int index = op.getDeviceNames().size(); for (auto use : op.getOperation()->getUsers()) { // Skip if it doesn't feed into case. Alternatively this could always // return the CPU device index if it exists.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0)