Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getDeviceNames (0.11 sec)

  1. 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)
  2. 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)
Back to top