- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for attrs (0.13 sec)
-
tensorflow/c/experimental/gradients/nn_grad.cc
}; // TODO(vnvo2409): Add python test class BiasAddGradientFunction : public GradientFunction { public: explicit BiasAddGradientFunction(AttrBuilder f_attrs) : forward_attrs_(f_attrs) {} Status Compute(AbstractContext* ctx, absl::Span<AbstractTensorHandle* const> grad_outputs, absl::Span<AbstractTensorHandle*> grad_inputs) override {
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad.cc
}; class MatMulGradientFunction : public GradientFunction { public: explicit MatMulGradientFunction(vector<AbstractTensorHandle*> f_inputs, AttrBuilder f_attrs) : forward_inputs_(f_inputs), forward_attrs_(f_attrs) { for (auto input : forward_inputs_) { if (input) { input->Ref(); } } } Status Compute(AbstractContext* ctx,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 15.2K bytes - Viewed (0) -
tensorflow/c/c_api.cc
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/BUILD
"//tensorflow/core:lib", "@com_google_absl//absl/memory", ], ) cc_library( name = "tfe_op_attrs_internal", hdrs = ["tfe_op_attrs_internal.h"], visibility = [ "//tensorflow:internal", ], deps = [ ":abstract_op_attrs", "//tensorflow/c:conversion_macros", "//tensorflow/c:tf_status", "//tensorflow/core:protos_all_cc", ],
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 33.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) { tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs)); } void TFE_OpAttrsSerialize(const TFE_OpAttrs* attrs, TF_Buffer* buf, TF_Status* status) { tensorflow::NameAttrList name_and_attrs; tensorflow::unwrap(attrs)->GetNameAttrList(&name_and_attrs); status->status = MessageToBuffer(name_and_attrs, buf); } namespace tensorflow {
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/eager/c_api_experimental.h
// while `op` is alive. TF_CAPI_EXPORT extern const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op); // Add attributes in `attrs` to `op`. // // Does not overwrite or update existing attributes, but adds new ones. TF_CAPI_EXPORT extern void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs); // Serialize `attrs` as a tensorflow::NameAttrList protocol buffer (into `buf`), // containing the op name and a map of its attributes.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
Status SetAttrFloat(const char* attr_name, float value) override { op_->node_builder.Attr(attr_name, value); 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(
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_api_function.cc
return; } (*(fdef_or.value()->mutable_attr()))[string(attr_name)] = attr_value; status->status = absl::OkStatus(); } void TF_FunctionGetAttrValueProto(TF_Function* func, const char* attr_name, TF_Buffer* output_attr_value, TF_Status* status) { const auto& it = func->record->fdef().attr().find(attr_name); if (it == func->record->fdef().attr().end()) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 13.6K bytes - Viewed (2) -
tensorflow/c/c_api_experimental.cc
void TF_AttrBuilderSetType(TF_AttrBuilder* builder, const char* attr_name, TF_DataType value) { auto iter = builder->attr_names.insert(attr_name).first; builder->Set(*iter, static_cast<tensorflow::DataType>(value)); } void TF_AttrBuilderSetTypeList(TF_AttrBuilder* builder, const char* attr_name, const TF_DataType* values, int num_values) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
ATTR_TEST_REGISTER_OP(string); ATTR_TEST_REGISTER_OP(int); ATTR_TEST_REGISTER_OP(float); ATTR_TEST_REGISTER_OP(bool); ATTR_TEST_REGISTER_OP(type); ATTR_TEST_REGISTER_OP(shape); ATTR_TEST_REGISTER_OP(tensor); #undef ATTR_TEST_REGISTER_OP class CApiAttributesTest : public ::testing::Test { protected: CApiAttributesTest()
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)