Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_OpSetAttrType (0.16 sec)

  1. 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);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 25 15:21:13 GMT 2023
    - 12.9K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api.h

    TF_CAPI_EXPORT extern void TFE_OpSetAttrBool(TFE_Op* op, const char* attr_name,
                                                 unsigned char value);
    TF_CAPI_EXPORT extern void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name,
                                                 TF_DataType value);
    // If the number of dimensions is unknown, `num_dims` must be set to
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 22.8K bytes
    - Viewed (1)
Back to top