Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_NewAttrBuilder (0.19 sec)

  1. tensorflow/c/c_api_experimental.h

    // TF_NewAttrBuilder() returns an object that you can set attributes on as
    // though it were an op. This allows querying properties of that op for
    // type-checking purposes like if the op will run on a particular device type.
    typedef struct TF_AttrBuilder TF_AttrBuilder;
    TF_CAPI_EXPORT extern TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name);
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Apr 27 21:07:00 GMT 2023
    - 15.1K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.cc

      // `builder->Set()` will outlive the subsequent
      // `TF_AttrBuilderCheckCanRunOnDevice()` call(s) on the same `builder`.
      std::set<std::string> attr_names;
    };
    
    TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name) {
      return new TF_AttrBuilder(op_name);
    }
    
    void TF_DeleteAttrBuilder(TF_AttrBuilder* builder) { delete builder; }
    
    void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
Back to top