Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 19 for from (0.15 sec)

  1. tensorflow/c/experimental/grappler/grappler.cc

              ->LookUpOpDef(name, &op_def_ptr);
      if (!s.ok()) {
        tsl::Set_TF_Status_from_Status(status, s);
        return;
      }
    
      s = tensorflow::MessageToBuffer(*op_def_ptr, buf);
      if (!s.ok()) {
        tsl::Set_TF_Status_from_Status(status, s);
        return;
      }
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  2. 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"
    #include "tensorflow/c/tf_status_internal.h"
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device.cc

        const TFE_OpAttrs* attributes, int expected_max_outputs,
        TF_Status* status) {
      absl::optional<std::vector<MaybeParallelTensorOwned>> result;
      // TODO(allenl): We should remove "TPU" from these op names at the very least,
      // or consider other ways of packing/unpacking parallel tensors.
      if (operation_name == std::string("TPUReplicatedInput")) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  4. tensorflow/c/c_api_experimental.cc

      opts->opts.validate_colocation_constraints = enable;
    }
    
    // Load a Pluggable Device library.
    // On success, returns the handle to library in result and return OK from the
    // function. Otherwise return nullptr in result and error Status from the
    // function.
    //
    // If `library_filename` has already been loaded, we return a cached handle.
    // Device and Kernels/Ops are registered as globals when a library is loaded
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_function.cc

      tensorflow::FunctionDef fdef;
      status->status = tensorflow::GraphToFunctionDef(
          fn_body->graph, fn_name, append_hash_to_fn_name != 0,
          /*set_stateful_from_nodes=*/true,
          /*copy_placeholder_attrs_from_nodes=*/true, body_nodes, input_tensors,
          output_tensors, output_names_vec, control_output_nodes,
          control_output_names_vec, description, &fdef);
      if (TF_GetCode(status) != TF_OK) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 13.6K bytes
    - Viewed (2)
  6. 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,
        std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Jun 15 15:44:44 GMT 2021
    - 12.5K bytes
    - Viewed (0)
  7. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

                        CancellationManager& cancellation_manager,
                        absl::optional<int64_t> step_id = absl::nullopt);
      // Block until the previous `StartExecute` operation has executed. Forwards
      // the status from `TFE_Execute` and returns outputs if the status is OK.
      std::vector<TensorHandlePtr> Join(TF_Status* status);
    
      // Block until all Ops finished running on the thread.
      void AsyncWait(TF_Status* status);
    
     private:
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  8. tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc

        return nullptr;
      else
        return &tmp_dir;
    }
    
    namespace tensorflow {
    namespace {
    
    // TODO(vnvo2409): Refactor `gcs_filesystem_test` to remove unnecessary tests
    // after porting all tests from
    // `//tensorflow/core/platform/cloud:gcs_file_system_test`.
    class GCSFilesystemTest : public ::testing::Test {
     public:
      void SetUp() override {
        root_dir_ = io::JoinPath(
            *GetTmpDir(),
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Aug 31 12:04:23 GMT 2020
    - 24.9K bytes
    - Viewed (0)
  9. tensorflow/c/experimental/filesystem/plugins/posix/posix_filesystem.cc

      ops->filesystem_ops->new_writable_file = tf_posix_filesystem::NewWritableFile;
      ops->filesystem_ops->new_appendable_file =
          tf_posix_filesystem::NewAppendableFile;
      ops->filesystem_ops->new_read_only_memory_region_from_file =
          tf_posix_filesystem::NewReadOnlyMemoryRegionFromFile;
      ops->filesystem_ops->create_dir = tf_posix_filesystem::CreateDir;
      ops->filesystem_ops->delete_file = tf_posix_filesystem::DeleteFile;
    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)
  10. tensorflow/c/experimental/next_pluggable_device/c_api.cc

      absl::Status cc_status;
      if (arg_tensor.dtype() != tensorflow::DT_RESOURCE) {
        cc_status = absl::InvalidArgumentError(
            absl::StrCat("Trying to obtain resource handle from Input[", index,
                         "], which is not type DT_RESOURCE."));
        status->status = cc_status;
        return nullptr;
      }
      const tensorflow::ResourceHandle& handle =
    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)
Back to top