Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SetAttrBool (0.16 sec)

  1. tensorflow/c/eager/gradients.cc

      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrFloat(attr_name, value);
    }
    Status SetAttrBool(AbstractOperation* op_, const char* attr_name, bool value,
                       ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrBool(attr_name, value);
    }
    Status SetAttrType(AbstractOperation* op_, const char* attr_name,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        return absl::OkStatus();
      }
      Status SetAttrFloat(const char* attr_name, float value) override {
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      Status SetAttrBool(const char* attr_name, bool value) override {
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      Status SetAttrType(const char* const attr_name, DataType value) override {
        if (!op_) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  3. tensorflow/c/c_api_test.cc

        TF_OperationDescription* desc = TF_NewOperation(graph, "MatMul", name);
        if (transpose_a) {
          TF_SetAttrBool(desc, "transpose_a", 1);
        }
        if (transpose_b) {
          TF_SetAttrBool(desc, "transpose_b", 1);
        }
        TF_AddInput(desc, {l, 0});
        TF_AddInput(desc, {r, 0});
        TF_Operation* op = TF_FinishOperation(desc, s);
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  4. tensorflow/c/eager/c_api.cc

      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrBool(TFE_Op* op, const char* attr_name, unsigned char value) {
      auto s = tensorflow::unwrap(op)->SetAttrBool(attr_name,
                                                   (value == 0) ? false : true);
      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)
  5. tensorflow/c/c_api.cc

                             const float* values, int num_values) {
      desc->node_builder.Attr(attr_name,
                              ArraySlice<const float>(values, num_values));
    }
    
    void TF_SetAttrBool(TF_OperationDescription* desc, const char* attr_name,
                        unsigned char value) {
      desc->node_builder.Attr(attr_name, static_cast<bool>(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