- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for it (0.12 sec)
-
tensorflow/c/c_api_experimental.cc
if (!status->status.ok()) { TF_DeleteCheckpointReader(reader); return nullptr; } const auto& m = reader->GetVariableToDataTypeMap(); for (auto it = m.begin(); it != m.end(); ++it) reader->variable_list.push_back(it->first); std::sort(reader->variable_list.begin(), reader->variable_list.end()); return reader; } void TF_DeleteCheckpointReader(TF_CheckpointReader* reader) { delete reader; }
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/filesystem/plugins/posix/posix_filesystem.cc
} PosixFile; static void Cleanup(TF_RandomAccessFile* file) { auto posix_file = static_cast<PosixFile*>(file->plugin_file); close(posix_file->fd); // This would be safe to free using `free` directly as it is only opaque. // However, it is better to be consistent everywhere. plugin_memory_free(const_cast<char*>(posix_file->filename)); delete posix_file; } static int64_t Read(const TF_RandomAccessFile* file, uint64_t offset, size_t n,
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sun Mar 24 20:08:23 GMT 2024 - 15.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
TFE_ExecutorWaitForAllPendingNodes(executor, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); TFE_DeleteExecutor(executor); TF_DeleteStatus(status); } // Read the value of variable `var` and save it into `out_value`. void ReadVariable(TFE_Context* ctx, TFE_TensorHandle* var, TFE_TensorHandle** out_value) { TF_Status* status = TF_NewStatus(); TFE_Op* op = TFE_NewOp(ctx, "ReadVariableOp", status);
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/eager/dlpack.cc
#include "tensorflow/core/platform/logging.h" namespace tensorflow { namespace { // Managing context for the DLManagedTensor, will manage the lifetime of // DLManagedTensor. When calling DLManagedTensor::deleter, it will notify the // original framework of destruction, and this context will be deleted also. struct TfDlManagedTensorCtx { TensorReference reference; std::vector<int64_t> shape; std::vector<int64_t> strides;
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/gradients.cc
AbstractContext* ctx_; }; // Returns the number of elements in the gradient tensor. int64_t TapeVSpace::NumElements(AbstractTensorHandle* tensor) const { // TODO(srbs): It seems like this is used only for performance optimization // and not for correctness. The only downside of keeping this 1 seems to be // that the gradient accumulation is unbounded and we will never
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
absl::variant<ParallelTensor*, TFE_TensorHandle*>; // 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/c_api_function.cc
const auto& it = func->record->fdef().attr().find(attr_name); if (it == func->record->fdef().attr().end()) { status->status = InvalidArgument("Function '", func->record->fdef().signature().name(), "' has no attr named '", attr_name, "'."); return; } status->status = MessageToBuffer(it->second, output_attr_value); }
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_distributed_test.cc
} private: bool first_call_ = true; }; // After the graph pass is registered, it takes effect globally and can affect // other test cases. Define a static variable to switch it on and off. bool GraphErrorInjectionPass::enabled_ = false; // Test to ensure that a registered graph optimization pass is only executed
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/c_api.cc
static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); } // Note: this function looks up a thread local policy. So it should be called in // the appropriate client thread. In particular, in async mode, it may not be // safe to call this function from the async EagerExecutor threads. extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy( TFE_Context* ctx) {
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/experimental/filesystem/modular_filesystem_registration.cc
namespace tensorflow { // Checks that all schemes provided by a plugin are valid. // TODO(b/139060984): More validation could be done here, based on supported // charset, maximum length, etc. Punting it for later. static Status ValidateScheme(const char* scheme) { if (scheme == nullptr) return errors::InvalidArgument( "Attempted to register filesystem with `nullptr` URI scheme"); return OkStatus(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 07 22:08:43 GMT 2023 - 12.8K bytes - Viewed (0)