- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for SetAttrShape (0.28 sec)
-
tensorflow/c/eager/abstract_operation.h
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, const PartialTensorShape shape); virtual Status SetAttrFunction(const char* attr_name,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
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, const AbstractOperation* value) override;
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) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
} 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) { if (num_dims > TensorShape::MaxDimensions()) { return errors::InvalidArgument("Value specified for `", attr_name, "` has ",
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/eager/gradients_internal.h
ForwardOperation*); Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, ForwardOperation*); Status SetAttrShape(AbstractOperation*, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation*); Status SetAttrFunction(AbstractOperation*, const char* attr_name,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
"op_type and op_name must be specified before specifying attrs."); } op_->node_builder.Attr(attr_name, value); return absl::OkStatus(); } Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) override { PartialTensorShape shape; if (num_dims >= 0) { shape = PartialTensorShape(ArraySlice<int64_t>(
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) -
tensorflow/c/eager/gradients.cc
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, const int64_t* dims, const int num_dims, ForwardOperation* forward_op_) { if (num_dims > TensorShape::MaxDimensions()) {
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) -
tensorflow/c/c_test_util.cc
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); ASSERT_NE(*op, nullptr); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:52 GMT 2021 - 17.8K bytes - Viewed (2) -
tensorflow/c/eager/c_api.cc
} } void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims, const int num_dims, TF_Status* out_status) { out_status->status = tensorflow::unwrap(op)->SetAttrShape(attr_name, dims, num_dims); } void TFE_OpSetAttrFunction(TFE_Op* op, const char* attr_name, const TFE_Op* value) { auto s = tensorflow::unwrap(op)->SetAttrFunction(
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) -
tensorflow/c/c_api_test.cc
EXPECT_TRUE(std::equal(std::begin(list), std::end(list), std::begin(values))); } TEST_F(CApiAttributesTest, Shape) { // Unknown shape auto desc = init("shape"); TF_SetAttrShape(desc, "v", nullptr, -1); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1);
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) -
tensorflow/c/c_api.h
// Set `num_dims` to -1 to represent "unknown rank". Otherwise, // `dims` points to an array of length `num_dims`. `dims[i]` must be // >= -1, with -1 meaning "unknown dimension". TF_CAPI_EXPORT extern void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name, const int64_t* dims, int num_dims);
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3)