Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_AttrBuilderCheckCanRunOnDevice (0.32 sec)

  1. tensorflow/c/c_api_experimental.cc

      using tensorflow::AttrBuilder::AttrBuilder;
      // The string buffers to make sure that any `attr_name` we pass into
      // `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);
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 29.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_experimental.h

                                                         int num_values);
    
    // Checks the tensorflow::NodeDef built via the methods above to see if it can
    // run on device_type.
    TF_CAPI_EXPORT extern void TF_AttrBuilderCheckCanRunOnDevice(
        TF_AttrBuilder* builder, const char* device_type, TF_Status* status);
    
    // For argument number input_index, fetch the corresponding number_attr that
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 15.1K bytes
    - Viewed (0)
Back to top