Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SetAttrType (0.14 sec)

  1. tensorflow/c/eager/c_api.cc

      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()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  2. tensorflow/c/c_api.cc

      for (int i = 0; i < num_values; ++i) {
        b[i] = values[i];
      }
      desc->node_builder.Attr(attr_name,
                              ArraySlice<const bool>(b.get(), num_values));
    }
    
    void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name,
                        TF_DataType value) {
      desc->node_builder.Attr(attr_name, static_cast<DataType>(value));
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top