Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 15 for SetAttrType (0.21 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.h

      Status SetAttrInt(const char* attr_name, int64_t value) override;
      Status SetAttrFloat(const char* attr_name, float value) override;
      Status SetAttrBool(const char* attr_name, bool value) override;
      Status SetAttrType(const char* attr_name, DataType value) override;
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override;
      Status SetAttrFunction(const char* attr_name,
    C
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 23 23:12:39 GMT 2020
    - 3.7K bytes
    - Viewed (1)
  2. tensorflow/c/experimental/gradients/tape/tape_operation.cc

      forward_op_.attrs.Set(attr_name, value);
      return parent_op_->SetAttrBool(attr_name, value);
    }
    Status TapeOperation::SetAttrType(const char* attr_name, DataType value) {
      forward_op_.attrs.Set(attr_name, value);
      return parent_op_->SetAttrType(attr_name, value);
    }
    Status TapeOperation::SetAttrShape(const char* attr_name, const int64_t* dims,
                                       const int num_dims) {
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Tue Jun 07 01:53:35 GMT 2022
    - 9K bytes
    - Viewed (1)
  3. tensorflow/c/c_test_util.cc

                           TF_DataType dtype, const std::vector<int64_t>& dims,
                           TF_Operation** op) {
      TF_OperationDescription* desc = TF_NewOperation(graph, "Placeholder", name);
      TF_SetAttrType(desc, "dtype", dtype);
      if (!dims.empty()) {
        TF_SetAttrShape(desc, "shape", dims.data(), dims.size());
      }
      *op = TF_FinishOperation(desc, s);
      ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  4. tensorflow/c/eager/gradients_internal.h

                        ForwardOperation*);
    Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value,
                       ForwardOperation*);
    Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value,
                       ForwardOperation*);
    Status SetAttrShape(AbstractOperation*, const char* attr_name,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  5. tensorflow/c/eager/abstract_operation.h

      virtual Status SetAttrInt(const char* attr_name, int64_t value) = 0;
      virtual Status SetAttrFloat(const char* attr_name, float value) = 0;
      virtual Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
      virtual Status SetAttrShape(const char* attr_name, const int64_t* dims,
                                  const int num_dims) = 0;
      virtual Status SetAttrShape(const char* attr_name,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  6. 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 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. 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 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  8. 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 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  9. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT);
      TF_Operation* placeholder_op = TF_FinishOperation(placeholder_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
      TF_Output x{placeholder_op, 0};
    
      TF_OperationDescription* reduce_desc =
          TF_NewOperation(body.get(), "CollectiveReduce", "CollectiveReduce");
      TF_SetAttrType(reduce_desc, "T", TF_FLOAT);
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  10. tensorflow/c/eager/c_api_experimental_test.cc

          TF_NewOperation(function_graph, "Placeholder", "arg");
      TF_SetAttrType(arg_descr, "dtype", TF_INT32);
      TF_Status* status = TF_NewStatus();
      TF_Operation* arg = TF_FinishOperation(arg_descr, status);
      ASSERT_TRUE(TF_GetCode(status) == TF_OK) << TF_Message(status);
      TF_OperationDescription* id_descr =
          TF_NewOperation(function_graph, "Identity", "id");
      TF_SetAttrType(id_descr, "T", TF_INT32);
      TF_AddInput(id_descr, {arg, 0});
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 03:14:26 GMT 2023
    - 31.5K bytes
    - Viewed (1)
Back to top