Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/c/eager/parallel_device/parallel_device_lib.h

        if (TF_GetCode(status) != TF_OK) return nullptr;
        TFE_OpSetDevice(const_op.get(), underlying_devices_[device_index].c_str(),
                        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;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Mon Oct 21 04:14:14 UTC 2024
    - 13.1K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.cc

      auto s = tensorflow::unwrap(op)->SetAttrFunctionName(attr_name, data, length);
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrTensor(TFE_Op* op, const char* attr_name, TF_Tensor* tensor,
                             TF_Status* status) {
      tensorflow::Tensor t;
      status->status = TF_TensorToTensor(tensor, &t);
      tensorflow::TensorInterface interface(t);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top