- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 41 for init (0.14 sec)
-
tensorflow/c/eager/c_api_cluster_test.cc
#include "tensorflow/core/protobuf/tensorflow_server.pb.h" namespace { using ::tensorflow::string; void ReplaceTaskInServerDef(tensorflow::ServerDef* server_def, int task_index) { tensorflow::JobDef* job_def = server_def->mutable_cluster()->mutable_job(0); int port = tensorflow::testing::PickUnusedPortOrDie(); job_def->mutable_tasks()->at(task_index) = tensorflow::strings::StrCat("localhost:", port); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/math_grad_test.cc
{12, 15, 18, 12, 15, 18, 12, 15, 18}, {6, 6, 6, 15, 15, 15, 24, 24, 24}, {6, 15, 24, 6, 15, 24, 6, 15, 24}}; for (int i{}; i < 4; ++i) { bool transpose_a = transpose_a_vals[i]; bool transpose_b = transpose_b_vals[i]; Model MatMulModel = [transpose_a, transpose_b]( AbstractContext* ctx,
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) -
tensorflow/c/c_api_function.cc
const TF_Graph* fn_body, const char* fn_name, int ninputs, const TF_Output* inputs, std::vector<OutputTensor>* input_tensors, std::unordered_map<const Node*, std::vector<int>>* input_nodes) TF_EXCLUSIVE_LOCKS_REQUIRED(fn_body->mu) { input_tensors->reserve(ninputs); for (int i = 0; i < ninputs; ++i) { Node* node = inputs[i].oper ? &inputs[i].oper->node : nullptr; int idx = inputs[i].index;
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/eager/parallel_device/parallel_device_lib.cc
} TFE_OpAddAttrs(op_.get(), attributes); for (int input_index = 0; input_index < inputs.size(); ++input_index) { TFE_OpAddInput(op_.get(), inputs[input_index], status); if (TF_GetCode(status) != TF_OK) return; } std::vector<TFE_TensorHandle*> unwrapped_results(expected_max_outputs); int real_num_outputs = expected_max_outputs;
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) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
TFE_OpSetDevice(op.get(), device, status); if (TF_GetCode(status) != TF_OK) return; TFE_TensorHandle* result_handles[num_replicas]; int num_retvals = num_replicas; TFE_Execute(op.get(), result_handles, &num_retvals, status); if (TF_GetCode(status) != TF_OK) return; for (int i = 0; i < num_replicas; ++i) { (*components)[i].reset(result_handles[i]); } }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
reinterpret_cast<NamedParallelDevice*>(device_info); std::vector<MaybeParallelTensorUnowned> typed_inputs; int num_inputs = TFE_OpGetFlatInputCount(original_op, status); if (TF_GetCode(status) != TF_OK) return; typed_inputs.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { TFE_TensorHandle* input = TFE_OpGetFlatInput(original_op, i, status); if (TF_GetCode(status) != TF_OK) return;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
public: // Eager async execution is only supported when remote eager is not in use // (b/157523095). explicit ParallelDevice(const std::vector<std::string>& devices, bool is_async = false, int in_flight_nodes_limit = 0); ~ParallelDevice(); // Helper to copy a tensor handle from another device once for each component // of the ParallelDevice. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
// by the underlying tensorflow::VariableInfo. TF_VariableInfo(int index, const std::string& name, tensorflow::Var* var) { var_info = tensorflow::VariableInfo{index, name, var}; } tensorflow::VariableInfo var_info{0, "", nullptr}; }; TF_VariableInfo* TF_CreateVariableInfoFromContext(TF_OpKernelContext* ctx, int index,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
TFE_TensorHandle* result_handles; int num_retvals = 1; TFE_Execute(op.get(), &result_handles, &num_retvals, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_INVALID_ARGUMENT) << TF_Message(status.get()); } } TensorHandlePtr CollectiveSum(TFE_Context* context, TFE_TensorHandle* input, int group_size, TF_Status* status) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
// When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is // placed on task1, Handle types are LOCAL, REMOTE, LOCAL on task1. int num_replicas = 3; std::vector<TFE_TensorHandle*> handles = {h0, h1, h2}; TFE_TensorHandle* packed_handle = TFE_CreatePackedTensorHandle(ctx, handles.data(), &num_replicas, status);
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)