- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for index (0.22 sec)
-
tensorflow/c/c_api.cc
if (ic->num_outputs() <= new_src.index) { status->status = tensorflow::errors::OutOfRange( "Cannot update edge. Output index [", new_src.index, "] is greater than the number of total outputs [", ic->num_outputs(), "]."); return; } tensorflow::shape_inference::ShapeHandle shape = ic->output(new_src.index); tensorflow::shape_inference::InferenceContext* ic_dst =
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
int index) { DCHECK(index >= 0 && index < shape_list->num_items); TF_ShapeAndType& shape = shape_list->items[index]; DCHECK(shape.dims == nullptr) << "Shape at " << index << " is already set!"; shape.num_dims = -1; shape.dims = nullptr; } void TF_ShapeAndTypeListSetDtype(TF_ShapeAndTypeList* shape_list, int index, TF_DataType dtype) {
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/experimental/gradients/grad_test_helper.cc
if (!outputs[i]) continue; AbstractTensorHandlePtr numerical_grad; { AbstractTensorHandle* numerical_grad_raw; s = CalcNumericalGrad(ctx, model, inputs, /*input_index=*/i, use_function, &numerical_grad_raw); ASSERT_EQ(errors::OK, s.code()) << s.message(); numerical_grad.reset(numerical_grad_raw); } TF_Tensor* numerical_tensor;
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); std::vector<TFE_TensorHandle*> device_inputs; device_inputs.reserve(inputs.size()); for (int input_index = 0; input_index < inputs.size(); ++input_index) { // Parallel tensors are divided between operations by device.
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/c_api_distributed_test.cc
} void TestFunctionWithPackedInput(const bool remote) { tensorflow::ServerDef server_def = GetServerDef(3); // This server def has the task index set to 0. string serialized = server_def.SerializeAsString(); server_def.set_task_index(1); std::unique_ptr<tensorflow::GrpcServer> worker_server1; ASSERT_TRUE(tensorflow::GrpcServer::Create(
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/gradient_checker.cc
int input_index, bool use_function, AbstractTensorHandle** numerical_grad) { vector<AbstractTensorHandle*> theta_inputs(inputs.size()); for (int i{}; i < inputs.size(); ++i) { theta_inputs[i] = inputs[i]; } AbstractTensorHandle* theta = theta_inputs[input_index]; // parameter we are grad checking
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.cc
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/c_api_function.cc
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; TF_RETURN_WITH_CONTEXT_IF_ERROR( fn_body->graph.IsValidOutputTensor(node, idx), "Encountered while processing input ", i, " into function '", fn_name, "'");
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/c_api_test_util.cc
const tensorflow::ServerDef& server_def, int task_index) { tensorflow::ServerDef server_def_copy = server_def; tensorflow::ClusterDef* cluster_def = server_def_copy.mutable_cluster(); tensorflow::JobDef* job_def = cluster_def->mutable_job(0); const 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: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api.cc
return num_elements; } int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, TF_Status* status) { if (h == nullptr) { status->status = tensorflow::errors::InvalidArgument("Invalid handle"); return -1; } int64_t dim = -1; status->status = tensorflow::unwrap(h)->Dim(dim_index, &dim); return dim; }
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)