- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for Attrs (0.12 sec)
-
tensorflow/c/eager/abstract_op_attrs.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_ #define TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_ #include "absl/container/inlined_vector.h" #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed May 26 22:20:27 GMT 2021 - 2K bytes - Viewed (0) -
tensorflow/c/eager/tfe_op_attrs_internal.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_TFE_OP_ATTRS_INTERNAL_H_ #define TENSORFLOW_C_EAGER_TFE_OP_ATTRS_INTERNAL_H_ #include "tensorflow/c/conversion_macros.h" #include "tensorflow/c/eager/abstract_op_attrs.h" #include "tensorflow/c/tf_status.h" #include "tensorflow/core/framework/attr_value.pb.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 10 05:41:19 GMT 2021 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
forward_op_.attrs.Set(attr_name, value); return parent_op_->SetAttrFloat(attr_name, value); } Status TapeOperation::SetAttrBool(const char* attr_name, bool value) { forward_op_.attrs.Set(attr_name, value); return parent_op_->SetAttrBool(attr_name, value); } Status TapeOperation::SetAttrType(const char* attr_name, DataType value) { forward_op_.attrs.Set(attr_name, 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/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/eager/gradients.cc
} // TODO(b/166669239): This is needed to support AttrBuilder::Get for string // attributes. Number type attrs and DataType attrs work fine without this. // Consider getting rid of this and making the behavior between number types // and string consistent. forward_op_->attrs.BuildNodeDef(); std::unique_ptr<GradientFunction> gradient_fn; TF_RETURN_IF_ERROR(registry.Lookup(*forward_op_, &gradient_fn));
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/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/eager/c_api_test.cc
tensorflow::NameAttrList name_and_attrs; ASSERT_TRUE(name_and_attrs.ParseFromArray(serialized_attr_values->data, serialized_attr_values->length)); ASSERT_EQ("VarHandleOp", name_and_attrs.name()); ASSERT_EQ(tensorflow::DT_INT64, name_and_attrs.attr().find("dtype")->second.type()); TF_DeleteBuffer(serialized_attr_values);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_internal.h
#include "tensorflow/c/eager/tfe_executor_internal.h" // IWYU pragma: export #include "tensorflow/c/eager/tfe_monitoring_internal.h" // IWYU pragma: export #include "tensorflow/c/eager/tfe_op_attrs_internal.h" // IWYU pragma: export #include "tensorflow/c/eager/tfe_tensor_debug_info_internal.h" // IWYU pragma: export // TODO(b/154564140): Move this to its own header. This requires splitting // c_api_experimental.h
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jan 18 19:26:34 GMT 2023 - 1.9K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_operation.h
virtual void SetStackTrace(ManagedStackTrace stack_trace) = 0; virtual const tensorflow::AbstractOpAttrs* GetOpAttrs() const = 0; virtual void AddAttrs(const AbstractOpAttrs* op_attrs) = 0; virtual void SetCancellationManager( CancellationManager* cancellation_manager) = 0; // Returns the stack trace set by `SetStackTrace` if exists.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 22:40:32 GMT 2022 - 3.6K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} int TF_OperationGetNumAttrs(TF_Operation* oper) { return oper->node.attrs().size(); } int TF_OperationGetAttrNameLength(TF_Operation* oper, int i) { auto attrs = oper->node.attrs(); int count = 0; AttrValueMap::const_iterator it; for (it = attrs.begin(); it != attrs.end(); it++) { if (count == i) { return it->first.length(); } count++; }
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)