Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TFE_OpSetAttrType (0.11 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

      std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> handle_op(
          TFE_NewOp(context.get(), "VarHandleOp", status.get()), TFE_DeleteOp);
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get());
      TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT);
      TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0,
                         status.get());
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 15.6K bytes
    - Viewed (0)
  2. tensorflow/c/eager/parallel_device/parallel_device_lib.h

                        status);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetAttrTensor(const_op.get(), "value", tensor.get(), status);
        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetAttrType(const_op.get(), "dtype", datatype_enum);
        TFE_TensorHandle* device_handle;
        int num_outputs = 1;
        TFE_Execute(const_op.get(), &device_handle, &num_outputs, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api.cc

                                                   (value == 0) ? false : true);
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name, TF_DataType value) {
      auto s = tensorflow::unwrap(op)->SetAttrType(
          attr_name, static_cast<tensorflow::DataType>(value));
      if (!s.ok()) {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
  4. tensorflow/c/eager/c_api_cluster_test.cc

                      TFE_TensorHandle** out_value) {
      TF_Status* status = TF_NewStatus();
      TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      TFE_OpSetAttrType(op, "dtype", TF_FLOAT);
      TFE_OpAddInput(op, var, status);
      ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      int num_retvals = 1;
      TFE_Execute(op, out_value, &num_retvals, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.2K bytes
    - Viewed (0)
Back to top