- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for Builder (0.2 sec)
-
tensorflow/c/c_api.cc
Node* ret = nullptr; if (desc->graph->name_map.count(desc->node_builder.node_name())) { status->status = InvalidArgument("Duplicate node name in graph: '", desc->node_builder.node_name(), "'"); } else { if (!desc->colocation_constraints.empty()) { desc->node_builder.Attr( tensorflow::kColocationAttrName,
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/c_api_experimental.cc
} void TF_DeleteAttrBuilder(TF_AttrBuilder* builder) { delete builder; } 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,
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/eager/c_api_test.cc
void Compute(tensorflow::OpKernelContext* ctx) override { ctx->SetStatus(tensorflow::errors::Unavailable("Test error.")); } }; REGISTER_KERNEL_BUILDER( Name("TestNonCommUnavailable").Device(tensorflow::DEVICE_DEFAULT), TestUnavailableErrorOp); REGISTER_KERNEL_BUILDER( Name("TestCommUnavailable").Device(tensorflow::DEVICE_DEFAULT), TestUnavailableErrorOp);
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/c_api_test.cc
}; // Test we can query kernels REGISTER_OP("TestOpWithSingleKernel") .Input("a: float") .Input("b: float") .Output("o: float"); REGISTER_KERNEL_BUILDER( Name("TestOpWithSingleKernel").Device(tensorflow::DEVICE_CPU), DummyKernel); TEST(TestKernel, TestGetAllRegisteredKernels) { TF_Status* status = TF_NewStatus();
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) -
tensorflow/c/eager/gradients.cc
#include "tensorflow/c/eager/abstract_tensor_handle.h" #include "tensorflow/c/eager/c_api_unified_experimental_internal.h" #include "tensorflow/c/eager/gradients_internal.h" #include "tensorflow/core/common_runtime/eager/attr_builder.h" #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h" #include "tensorflow/core/platform/errors.h" namespace tensorflow { namespace gradients { namespace {
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
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 { op_->node_builder.Attr(attr_name, value); 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
#include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/device_factory.h" #include "tensorflow/core/common_runtime/device_mgr.h" #include "tensorflow/core/common_runtime/eager/attr_builder.h" #include "tensorflow/core/common_runtime/eager/context.h" #include "tensorflow/core/common_runtime/eager/custom_device.h" #include "tensorflow/core/common_runtime/eager/custom_device_op_handler.h"
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)