- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for SetAttrString (0.48 sec)
-
tensorflow/c/experimental/gradients/tape/tape_operation.cc
for (auto input : inputs) { forward_op_.inputs.push_back(input); } return OkStatus(); } Status TapeOperation::SetAttrString(const char* attr_name, const char* data, size_t length) { forward_op_.attrs.Set(attr_name, StringPiece(data, length)); return parent_op_->SetAttrString(attr_name, data, length); } Status TapeOperation::SetAttrInt(const char* attr_name, int64_t value) {
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
Status AddInputList(AbstractOperation*, absl::Span<AbstractTensorHandle* const> inputs, ForwardOperation*); // Sets the attrs in the `ForwardOperation`. Status SetAttrString(AbstractOperation*, const char* attr_name, const char* data, size_t length, ForwardOperation*); Status SetAttrInt(AbstractOperation*, const char* attr_name, int64_t value, ForwardOperation*);
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/abstract_operation.h
absl::Span<AbstractTensorHandle* const> inputs) = 0; virtual Status Execute(absl::Span<AbstractTensorHandle*> retvals, int* num_retvals) = 0; virtual Status SetAttrString(const char* attr_name, const char* data, size_t length) = 0; virtual Status SetAttrInt(const char* attr_name, int64_t value) = 0;
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/eager/gradients.cc
forward_op_->inputs.push_back(input); } return absl::OkStatus(); } Status SetAttrString(AbstractOperation* op_, const char* attr_name, const char* data, size_t length, ForwardOperation* forward_op_) { forward_op_->attrs.Set(attr_name, StringPiece(data, length)); return op_->SetAttrString(attr_name, data, length); }
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/gradients_test.cc
ASSERT_EQ(errors::OK, s.code()) << s.message(); } s = AddInput(check_numerics_op.get(), t.get(), &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); string message = "This is the way!"; s = SetAttrString(check_numerics_op.get(), "message", message.data(), message.length(), &forward_op); ASSERT_EQ(errors::OK, s.code()) << s.message(); int num_retvals = 1;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/nn_grad_test.cc
immediate_execution_ctx_.get(), {X.get(), Y.get()}, UseFunction())); } TEST_P(CppGradients, TestBiasAddGrad) { if (UseFunction() && UseMlir()) { GTEST_SKIP() << "SetAttrString has not been implemented yet.\n"; } // A float A_vals[] = {1.0f, 2.0f, 3.0f, 4.0f}; int64_t A_dims[] = {2, 2}; AbstractTensorHandlePtr A; { AbstractTensorHandle* A_raw;
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 8.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
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(), merge_op.length()); const std::string final_op("Id"); TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(), final_op.length()); TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0);
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/c_api_unified_experimental_graph.cc
*num_retvals = TF_OperationNumOutputs(operation); for (int i = 0; i < *num_retvals; ++i) { retvals[i] = new GraphTensor({operation, i}, g_); } return absl::OkStatus(); } Status SetAttrString(const char* attr_name, const char* data, size_t length) override { tensorflow::StringPiece s(data, length); op_->node_builder.Attr(attr_name, s); return absl::OkStatus(); }
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/c_api.cc
} TFE_DeleteOp(op); return ret; } void TFE_OpSetAttrString(TFE_Op* op, const char* attr_name, const void* value, size_t length) { auto s = tensorflow::unwrap(op)->SetAttrString( attr_name, static_cast<const char*>(value), length); 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) -
tensorflow/c/c_api_test.cc
EXPECT_EQ(expected_total_size, m.total_size); \ } while (0) TEST_F(CApiAttributesTest, String) { auto desc = init("string"); TF_SetAttrString(desc, "v", "bunny", 5); auto oper = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_); EXPECT_TF_META("v", -1, TF_ATTR_STRING, 5);
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)