- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for ruleset (1.8 sec)
-
tensorflow/c/eager/custom_device_test.cc
// Assign to the variable, copying to the custom device. std::unique_ptr<TFE_TensorHandle, decltype(&TFE_DeleteTensorHandle)> one( TestScalarTensorHandle(context.get(), 111.f), TFE_DeleteTensorHandle); op.reset(TFE_NewOp(context.get(), "AssignVariableOp", status.get())); TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); TFE_OpAddInput(op.get(), var_handle, status.get()); TFE_OpAddInput(op.get(), one.get(), status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 18.4K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
// These APIs are mainly to facilitate testing and are subject to change. namespace internal { Status Reset(AbstractOperation* op_, const char* op, const char* raw_device_name, ForwardOperation* forward_op_) { forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input,
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_experimental.h
#include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api_macros.h" #include "tensorflow/c/eager/c_api.h" #ifdef __cplusplus extern "C" { #endif // Resets `op_to_reset` with `op_or_function_name` and `raw_device_name`. This // is for performance optimization by reusing an exiting unused op rather than // creating a new op every time. If `raw_device_name` is `NULL` or empty, it
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/BUILD
# Description: # C API for TensorFlow, for use by client language bindings. load("@bazel_skylib//lib:selects.bzl", "selects") load("@local_config_tensorrt//:build_defs.bzl", "if_tensorrt") load("@local_tsl//tsl/platform:rules_cc.bzl", "cc_library") load( "//tensorflow:tensorflow.bzl", "check_deps", "if_google", "if_not_mobile", "tf_cc_test", "tf_copts", "tf_cuda_library", "tf_custom_op_library",
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 27 18:00:18 GMT 2024 - 30.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental_graph.cc
public: explicit GraphOperation(TF_Graph* g) : TracingOperation(kGraph), g_(g) {} void Release() override { delete this; } Status Reset(const char* op, const char* raw_device_name) override { if (op_) { return errors::FailedPrecondition("Reset called on already built op."); } if (raw_device_name) { device_name_ = raw_device_name; } op_type_ = op;
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_experimental.cc
using tensorflow::string; void TFE_OpReset(TFE_Op* op_to_reset, const char* op_or_function_name, const char* raw_device_name, TF_Status* status) { if (op_to_reset) { tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(op_to_reset); op->Clear(); status->status = op->Reset(op_or_function_name, raw_device_name); } else { TF_SetStatus(status, TF_INVALID_ARGUMENT,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache.h
[](TF_Thread* thread) { TF_JoinThread(thread); }) { if (max_staleness_ > 0) { TF_ThreadOptions thread_options; TF_DefaultThreadOptions(&thread_options); pruning_thread_.reset( TF_StartThread(&thread_options, "TF_prune_FBC", PruneThread, this)); } TF_VLog(1, "GCS file block cache is %s.\n", (IsCacheEnabled() ? "enabled" : "disabled")); }
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Aug 31 04:46:34 GMT 2020 - 10.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
"Invalid tensorflow.ServerDef protocol buffer"); return; } status->status = tensorflow::unwrap(ctx)->GetDistributedManager()->SetOrUpdateServerDef( server_def, /*reset_context=*/true, keep_alive_secs, init_timeout_in_ms, retries, clear_existing_contexts); #endif // !IS_MOBILE_PLATFORM } TF_CAPI_EXPORT extern void TFE_ContextUpdateServerDef(TFE_Context* ctx,
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/BUILD
) load("//tensorflow:tensorflow.default.bzl", "cc_header_only_library", "filegroup") load( "//tensorflow/core/platform:build_config_root.bzl", "tf_cuda_tests_tags", ) load("//tensorflow/core/platform:rules_cc.bzl", "cc_library") package( # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"], licenses = ["notice"], ) tf_cuda_library( name = "c_api", srcs = [
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/experimental/gradients/math_grad_test.cc
ASSERT_EQ(errors::OK, status_.code()) << status_.message(); x.reset(x_raw); } AbstractTensorHandlePtr y; { AbstractTensorHandle* y_raw = nullptr; status_ = TestScalarTensorHandle<float, TF_FLOAT>( immediate_execution_ctx_.get(), 2.0f, &y_raw); ASSERT_EQ(errors::OK, status_.code()) << status_.message(); y.reset(y_raw); }
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Thu Apr 13 17:32:14 GMT 2023 - 16.3K bytes - Viewed (0)