- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for arpack (0.15 sec)
-
tensorflow/c/eager/dlpack.cc
limitations under the License. ==============================================================================*/ #include "tensorflow/c/eager/dlpack.h" #include <string> #include "include/dlpack/dlpack.h" // from @dlpack #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/tfe_tensorhandle_internal.h"
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) -
tensorflow/c/eager/c_api_experimental.h
// custom device. // // Many devices will want to simply return an "unimplemented" status // here. This is the default behavior if `pack` is null when passed to // TFE_RegisterCustomDevice. TFE_TensorHandle* (*pack)(TFE_Context* context, TFE_TensorHandle** handles, int num_handles, TF_Status* s, void* device_info) = nullptr;
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/BUILD
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 33.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
TFE_DeleteTensorHandle(outputs[i]); } } return status.status; } tensorflow::Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles, ImmediateExecutionTensorHandle** result) override { TF_Status status; *result = tensorflow::unwrap(device_.pack(context_, tensorflow::wrap(handles.data()),
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/eager/c_api_experimental.cc
// One of the inputs we're trying to pack is on a custom device. We'll let // the first custom device we see handle all of the packing. auto* custom_device_handle = tensorflow::down_cast<tensorflow::CustomDeviceTensorHandle*>( unwrapped_handle); tensorflow::ImmediateExecutionTensorHandle* result; status->status = custom_device_handle->device()->Pack(
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/c_api_experimental.cc
if (!status->status.ok()) return; // Initialize a input_tensor vector with `nullptr` values. std::vector<const Tensor*> input_tensors_vector(num_inputs, nullptr); // A vector to keep track of newly created `tf::Tensor` objects. std::vector<Tensor> all_input_tensors; // Update the vector with information from `input_tensors` if provided. if (input_tensors != nullptr) {
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/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.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)