- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for arpack (0.2 sec)
-
tensorflow/c/eager/dlpack_test.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/dlpack.h" #include <vector> #include "absl/strings/str_join.h" #include "include/dlpack/dlpack.h" // from @dlpack #include "tensorflow/c/eager/c_api.h" #include "tensorflow/core/platform/test.h" namespace tensorflow { namespace {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
#include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { // Abstract interface to a TensorHandle. // // A TensorHandle is management class around a Tensor which may track additional // metadata and synchronization. // // This allows us to hide concrete implementations of TensorHandle from header // files. The interface lists the common functionality that must be provided by
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0) -
tensorflow/c/c_api.cc
"names (", n->name(), "). This is a bug. Please file an issue at " "https://github.com/tensorflow/tensorflow/issues."); } } } // Unpack the results from grad_outputs_arg. TFOutputsFromOutputs(dy_arg, dy); #endif // defined(IS_MOBILE_PLATFORM) || defined(IS_SLIM_BUILD) } // TF_Session functions ----------------------------------------------
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/experimental/next_pluggable_device/c_api.cc
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_testlib.h
TF_Status* 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, std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status); // Helper to pack `num_replicas` TFE_TensorHandles into one parallel handle. template <std::size_t num_replicas>
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// A ParallelDevice on its own is not registered with a TFE_Context, and so has // no device name (e.g. for `tf.device`). `NamedParallelDevice` associates a // name with it, which lets us pack its `ParallelTensor`s into TFE_TensorHandles // placed on the parallel device. class NamedParallelDevice { public: NamedParallelDevice(const std::string& name,
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
if (first_bad_status != nullptr) { TF_SetStatus(status, TF_GetCode(first_bad_status.get()), TF_Message(first_bad_status.get())); return result; } // For each output of the original operation, pack the per-device // TensorHandles we've computed into a single parallel TensorHandle. std::vector<std::unique_ptr<ParallelTensor>> per_device_outputs; per_device_outputs.reserve(first_op_output_count);
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)