- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 34 for pack (0.15 sec)
-
tensorflow/c/c_api.cc
opts->opts.control_dependencies.push_back(oper->node.name()); } void TF_ImportGraphDefOptionsAddReturnOutput(TF_ImportGraphDefOptions* opts, const char* oper_name, int index) { opts->tensor_id_data.push_back(oper_name); const string& oper_name_str = opts->tensor_id_data.back(); opts->opts.return_tensors.emplace_back(oper_name_str, index); }
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/eager/c_api_unified_experimental.cc
return wrap(unwrap(o)->outputs[i]); } void TF_OutputListPushBack(TF_OutputList* o, TF_AbstractTensor* tensor, TF_Status* s) { unwrap(o)->outputs.push_back(unwrap(tensor)); } void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
->DebugString(), " as input to a parallel operation. First pack non-parallel " "tensors for each device into a parallel tensor explicitly.") .c_str()); return absl::nullopt; } } else { parallel_inputs.push_back(absl::get<ParallelTensor*>(input)); } } absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
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_test.cc
Multiply(context.get(), second_combined_value.get(), second_negative_one.get(), status.get())); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); // Un-pack the parallel tensor to verify that the operation was // successful. The resulting structure should be: // second_device{first_device{1. * 3., 2. * 3.}, 3. * 3.}. std::array<TensorHandlePtr, 2> second_components;
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/parallel_device/parallel_device_testlib.cc
&FloatDeallocator, nullptr), TF_DeleteTensor); return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); } // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle. template <std::size_t num_replicas> void ExtractPerDeviceValues( TFE_Context* context, TFE_TensorHandle* input,
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_lib.cc
components.reserve(underlying_devices_.size()); for (int j = 0; j < underlying_devices_.size(); ++j) { components.push_back(std::move(per_device_output_tensors[j][i])); } if (expected_output_shapes[i].IsFullyDefined()) { per_device_outputs.push_back(ParallelTensor::FromTensorHandles( *this, std::move(components), absl::Span<const int64_t>(expected_output_shapes[i].dim_sizes()),
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
// Add a sync point to make sure that variables have been initialized // before the function execution starts. TFE_ContextAsyncWait(ctx, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Pack 3 variable handles into one TFE_TensorHandle. // 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
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/parallel_device/parallel_device_lib_test.cc
ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); std::vector<TensorHandlePtr> vector_handles; vector_handles.reserve(2); vector_handles.push_back(std::move(two_vector)); vector_handles.push_back(std::move(three_vector)); std::unique_ptr<ParallelTensor> unknown_length_vector = ParallelTensor::FromTensorHandles( parallel_device, std::move(vector_handles), status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
all_input_tensors.reserve(num_inputs); for (int i = 0; i < num_inputs; ++i) { if (input_tensors[i] == nullptr) continue; all_input_tensors.emplace_back(); Tensor& input_tensor = all_input_tensors.back(); status->status = TF_TensorToTensor(input_tensors[i], &input_tensor); if (!status->status.ok()) return; input_tensors_vector[i] = &input_tensor; } }
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/c_api_function.cc
for (int i = 0; i < noutputs; ++i) { output_names_vec.push_back(string(output_names[i])); } } // Process control output names. std::vector<string> control_output_names_vec; if (control_output_names) { control_output_names_vec.reserve(ncontrol_outputs); for (int i = 0; i < ncontrol_outputs; ++i) { control_output_names_vec.push_back(string(control_output_names[i])); } }
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)