- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for SetAttrShape (0.15 sec)
-
tensorflow/c/eager/abstract_operation.h
virtual absl::Status SetAttrType(const char* attr_name, DataType value) = 0; virtual absl::Status SetAttrShape(const char* attr_name, const int64_t* dims, const int num_dims) = 0; virtual absl::Status SetAttrShape(const char* attr_name, const PartialTensorShape shape); virtual absl::Status SetAttrFunction(const char* attr_name,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
ForwardOperation*); absl::Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value, ForwardOperation*); absl::Status SetAttrShape(AbstractOperation*, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation*); absl::Status SetAttrFunction(AbstractOperation*, const char* attr_name,
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 4.7K 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(); } absl::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>(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
DataType value, ForwardOperation* forward_op_) { forward_op_->attrs.Set(attr_name, value); return op_->SetAttrType(attr_name, value); } absl::Status SetAttrShape(AbstractOperation* op_, const char* attr_name, const int64_t* dims, const int num_dims, ForwardOperation* forward_op_) { if (num_dims > TensorShape::MaxDimensions()) {
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0) -
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(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0)