Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for SetAttrType (0.2 sec)

  1. tensorflow/c/eager/c_api_unified_experimental_graph.cc

        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_) {
          return Status(
              absl::StatusCode::kFailedPrecondition,
              "op_type and op_name must be specified before specifying attrs.");
    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)
  2. tensorflow/c/eager/gradients.cc

      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrBool(attr_name, value);
    }
    Status SetAttrType(AbstractOperation* op_, const char* attr_name,
                       DataType value, ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrType(attr_name, value);
    }
    Status SetAttrShape(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)
  3. tensorflow/c/eager/c_api_unified_experimental.cc

    }
    
    void TF_AbstractOpSetAttrType(TF_AbstractOp* op, const char* const attr_name,
                                  TF_DataType value, TF_Status* s) {
      Status status =
          unwrap(op)->SetAttrType(attr_name, static_cast<DataType>(value));
      TF_SetStatus(s, static_cast<TF_Code>(status.code()),
                   tsl::NullTerminatedMessage(status));
    }
    
    void TF_ExecuteOperation(TF_AbstractOp* op, int num_inputs,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/c/c_api_test.cc

        TF_OperationDescription* desc = TF_NewOperation(graph, "Const", name);
        TF_SetAttrTensor(desc, "value", tensor.get(), s);
        if (TF_GetCode(s) != TF_OK) return nullptr;
        TF_SetAttrType(desc, "dtype", TF_FLOAT);
        TF_Operation* op = TF_FinishOperation(desc, s);
        EXPECT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        return op;
      }
    
    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)
  6. 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