Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for TensorHandleConvertible (0.21 sec)

  1. tensorflow/c/experimental/saved_model/core/revived_types/tensorhandle_convertible.h

    class TensorHandleConvertible {
     public:
      explicit TensorHandleConvertible(ImmediateTensorHandlePtr handle)
          : handle_(std::move(handle)) {}
    
      ImmediateExecutionTensorHandle* handle() { return handle_.get(); }
    
      // TensorHandleConvertible is movable, but not copyable.
      TensorHandleConvertible(TensorHandleConvertible&& other) = default;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 30 19:43:25 UTC 2022
    - 2.1K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/saved_model/core/tf_concrete_function_loading_test.cc

     private:
      std::unique_ptr<StaticDeviceMgr> device_mgr_;
      EagerContextPtr ctx_;
    };
    
    class DummyCapture : public TensorHandleConvertible {
     public:
      DummyCapture(ImmediateExecutionContext* ctx, int8_t value)
          : TensorHandleConvertible(
                testing::CreateTensorHandle(ctx, DT_FLOAT, {2, 4}, value)) {}
    };
    
    FunctionDef FuncDefWithNumInputsOutputs(int num_inputs, int num_outputs) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 10.6K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/saved_model/core/revived_types/constant.h

    // and offer a subclass of TensorHandleConvertible. Note that similar to
    // the python's eager mode logic, we bypass calling the "Const" op:
    // https://github.com/tensorflow/tensorflow/blob/1c064ab76064c58e54261b805027474885a1534d/tensorflow/python/framework/constant_op.py#L301
    class Constant : public TensorHandleConvertible {
     public:
      static Status Create(ImmediateExecutionContext* ctx,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 30 19:43:25 UTC 2022
    - 2.3K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/saved_model/core/revived_types/restored_resource.cc

                                       TFConcreteFunction* initialize,
                                       TFConcreteFunction* destroy_resource,
                                       ImmediateTensorHandlePtr resource_handle)
        : TensorHandleConvertible(std::move(resource_handle)),
          device_(device),
          create_resource_(create_resource),
          initialize_(initialize),
          destroy_resource_(destroy_resource) {}
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 3.1K bytes
    - Viewed (0)
  5. tensorflow/c/experimental/saved_model/core/revived_types/asset.cc

    #include "tensorflow/cc/saved_model/constants.h"
    #include "tensorflow/core/platform/errors.h"
    #include "tensorflow/core/platform/path.h"
    
    namespace tensorflow {
    
    Asset::Asset(ImmediateTensorHandlePtr handle)
        : TensorHandleConvertible(std::move(handle)) {}
    
    Status Asset::Create(ImmediateExecutionContext* ctx,
                         const std::string& saved_model_dir,
                         const std::string& asset_filename,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 09 20:11:48 UTC 2023
    - 1.9K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/saved_model/core/revived_types/variable.h

    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    #include "tensorflow/core/protobuf/saved_object_graph.pb.h"
    
    namespace tensorflow {
    
    class Variable : public TensorHandleConvertible {
     public:
      // Creates an uninitialized resource variable. Note that a caller must
      // call "assign" to associate a value with the variable.
      static Status CreateUninitialized(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 30 19:43:25 UTC 2022
    - 2.8K bytes
    - Viewed (0)
  7. tensorflow/c/experimental/saved_model/core/saved_model_utils.h

    Status LoadTFConcreteFunction(
        const SavedConcreteFunction& saved_concrete_function,
        const FunctionDef* function_def,
        const std::unordered_map<int, std::unique_ptr<TensorHandleConvertible>>&
            captured_objects,
        ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out);
    
    // Flattens `signature` into a vector of TensorSpecProto pointers back into
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Dec 13 04:18:52 UTC 2022
    - 5.7K bytes
    - Viewed (0)
  8. tensorflow/c/experimental/saved_model/core/saved_model_utils.cc

      return Status();
    }
    
    Status LoadTFConcreteFunction(
        const SavedConcreteFunction& saved_concrete_function,
        const FunctionDef* function_def,
        const std::unordered_map<int, std::unique_ptr<TensorHandleConvertible>>&
            captured_objects,
        ImmediateExecutionContext* ctx, std::unique_ptr<TFConcreteFunction>* out) {
      TF_RETURN_IF_ERROR(ValidateSavedFunctionCompatibleWithFunctionDef(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jan 12 19:17:46 UTC 2023
    - 24K bytes
    - Viewed (0)
Back to top