- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for options (0.17 sec)
-
tensorflow/c/eager/c_api_test_util.cc
tensorflow::strings::StrCat("/job:", job_name, "/replica:0/task:", 0)); auto* rewrite_options = config->mutable_graph_options()->mutable_rewrite_options(); rewrite_options->set_scoped_allocator_optimization( tensorflow::RewriterConfig::ON); rewrite_options->mutable_scoped_allocator_opts()->add_enable_op( "CollectiveReduce");
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api.cc
void TFE_ContextOptionsSetConfig(TFE_ContextOptions* options, const void* proto, size_t proto_len, TF_Status* status) { TF_SetConfig(&options->session_options, proto, proto_len, status); } void TFE_ContextOptionsSetAsync(TFE_ContextOptions* options, unsigned char enable) { options->async = enable; } void TFE_ContextOptionsSetDevicePlacementPolicy(
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_experimental.cc
return static_cast<TF_Operation*>(static_cast<void*>(node)); } void TF_EnableXLACompilation(TF_SessionOptions* options, unsigned char enable) { tensorflow::ConfigProto& config = options->options.config; auto* optimizer_options = config.mutable_graph_options()->mutable_optimizer_options(); if (enable) { optimizer_options->set_global_jit_level(tensorflow::OptimizerOptions::ON_1);
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_experimental.cc
if (prefix == nullptr || strlen(prefix) == 0) prefix = "/job:localhost/replica:0/task:0"; tensorflow::SessionOptions sess_options; (*sess_options.config.mutable_device_count())["CPU"] = num_cpus; status->status = tensorflow::DeviceFactory::AddCpuDevices(sess_options, prefix, &devices); // Remove the device that has the host device name since host device is alreay // in an initialized context.
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/BUILD
], deps = [ ":c_api", ":c_api_experimental", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core:session_options", "//tensorflow/core:test", ], ) tf_cc_test( name = "c_test", srcs = ["c_test.c"], extra_copts = ["-std=c11"], deps = [ ":c_api",
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_distributed_test.cc
class GraphErrorInjectionPass : public tensorflow::GraphOptimizationPass { public: static bool enabled_; GraphErrorInjectionPass() {} tensorflow::Status Run( const tensorflow::GraphOptimizationPassOptions& options) override { if (!enabled_) { return absl::OkStatus(); } if (first_call_) { first_call_ = false; return absl::OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
// There are two ways to represent compact row-major data // 1) nullptr indicates tensor is compact and row-majored. // 2) fill in the strides array as the real case for compact row-major data. // Here we choose option 2, since some frameworks didn't handle the strides // argument properly. dlm_tensor->dl_tensor.strides = stride_arr->data(); dlm_tensor->dl_tensor.byte_offset =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
CONTRIBUTING.md
`--per_file_copt` bazel option. For example, if you want to debug the Identity op, which are in files starting with `identity_op`, you can run ```bash bazel build --config=dbg --per_file_copt=+tensorflow/core/kernels/identity_op.*@-g //tensorflow/tools/pip_package:build_pip_package ```
Plain Text - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Thu Mar 21 11:45:51 GMT 2024 - 15.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// // Does not include the shape and dtype of the tensor (which is generally // appended later), but should include any information specific to this custom // device which would be useful for debugging. // // Optional. If null, defaults to resolving the TFE_TensorHandle into a // TF_Tensor and summarizing that. TF_Buffer* (*summarize)(void* data, TF_Status* status) = nullptr; } TFE_CustomDeviceTensorHandle;
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/parallel_device/parallel_device_lib.cc
TF_Message(first_bad_status.get())); } } absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> ParallelDevice::Join( const std::vector<PartialTensorShape>& expected_output_shapes, TF_Status* status) const { absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> result; // Compute per-device per-output tensors
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1)