- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for SetAttrInt (0.16 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TF_OperationDescription* reduce_desc = TF_NewOperation(body.get(), "CollectiveReduce", "CollectiveReduce"); TF_SetAttrType(reduce_desc, "T", TF_FLOAT); TF_SetAttrInt(reduce_desc, "group_size", group_size); TF_SetAttrInt(reduce_desc, "group_key", 0); TF_SetAttrInt(reduce_desc, "instance_key", 0); const std::string merge_op("Mul"); TF_SetAttrString(reduce_desc, "merge_op", merge_op.c_str(),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
return op_->SetAttrString(attr_name, data, length); } Status SetAttrInt(AbstractOperation* op_, const char* attr_name, int64_t value, ForwardOperation* forward_op_) { forward_op_->attrs.Set(attr_name, static_cast<int64_t>(value)); return op_->SetAttrInt(attr_name, value); } Status SetAttrFloat(AbstractOperation* op_, const char* attr_name, float value,
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/eager/c_api_unified_experimental_graph.cc
size_t length) override { tensorflow::StringPiece s(data, length); op_->node_builder.Attr(attr_name, s); return absl::OkStatus(); } Status SetAttrInt(const char* attr_name, int64_t value) override { op_->node_builder.Attr(attr_name, static_cast<int64_t>(value)); return absl::OkStatus(); } Status SetAttrFloat(const char* attr_name, float value) override {
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/c_test_util.cc
0, graph, s, ::tensorflow::strings::StrCat(name, "_const0").c_str()); TF_OperationDescription* desc = TF_NewOperation(graph, "Split", name); TF_AddInput(desc, {zero, 0}); TF_AddInput(desc, {input, 0}); TF_SetAttrInt(desc, "num_split", 3); TF_SetAttrType(desc, "T", TF_INT32); // Set device to CPU since there is no version of split for int32 on GPU // TODO(iga): Convert all these helpers and tests to use floats because
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
if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { auto s = tensorflow::unwrap(op)->SetAttrInt(attr_name, value); if (!s.ok()) { LOG(WARNING) << "Unable to set attribute: " << attr_name; } } void TFE_OpSetAttrFloat(TFE_Op* op, const char* attr_name, float value) {
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)