Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_SetDevice (0.05 sec)

  1. tensorflow/c/c_api_function_test.cc

        for (auto input : inputs) {
          TF_AddInput(desc, input);
        }
        // Set device to CPU because some ops inside the function might not be
        // available on GPU.
        TF_SetDevice(desc, "/cpu:0");
        *op = TF_FinishOperation(desc, s_);
        ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
        ASSERT_NE(*op, nullptr);
      }
    
      FunctionDef fdef() {
        tensorflow::FunctionDef fdef;
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Mon Nov 17 00:00:38 UTC 2025
    - 63.6K bytes
    - Viewed (1)
  2. tensorflow/c/c_api_test.cc

        TF_TString_Copy(&data[i], source.c_str(), source.length());
      }
    
      TF_DeleteTensor(t);
    }
    
    }  // namespace
    }  // namespace tensorflow
    
    // TODO(josh11b): Test:
    // * TF_SetDevice(desc, "/job:worker");
    // * control inputs / outputs
    // * targets
    Registered: Tue Dec 30 12:39:10 UTC 2025
    - Last Modified: Mon Nov 17 00:00:38 UTC 2025
    - 97K bytes
    - Viewed (0)
Back to top