Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for function (0.19 sec)

  1. tensorflow/c/c_api_function.cc

      for (const Node* n : fn_body->graph.nodes()) {
        stack_traces[n->name()] = n->GetStackTrace();
      }
    
      TF_Function* tf_function = new TF_Function();
      tf_function->record = new tensorflow::FunctionRecord(
          std::move(fdef), std::move(stack_traces), false);
    
      return tf_function;
    }
    
    TF_Function* TF_GraphToFunction(const TF_Graph* fn_body, const char* fn_name,
    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)
  2. tensorflow/c/eager/gradients.cc

    }
    
    // Calls the passed-in backward function.
    // op_type is the op's name provided in RecordOperation.
    Status TapeVSpace::CallBackwardFunction(
        const string& op_type, GradientFunction* gradient_function,
        const std::vector<int64_t>& unneeded_gradients,
        gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
        absl::Span<AbstractTensorHandle*> result) const {
      if (gradient_function == nullptr) {
    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)
  3. tensorflow/c/eager/c_api.cc

                                   TF_Status* status) {
      tensorflow::FunctionDef function_def;
      if (!function_def.ParseFromArray(serialized_function_def, size)) {
        status->status =
            tensorflow::errors::InvalidArgument("Invalid FunctionDef proto");
        return;
      }
    
      AnnotateEagerRuntimeConstructionContext(function_def);
    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)
  4. tensorflow/c/eager/c_api_experimental.h

                                                         const char* function_name,
                                                         TF_Buffer* buf,
                                                         TF_Status* status);
    
    // Get GraphDebugInfo containing stack traces mapping to node names
    TF_CAPI_EXPORT extern void TFE_ContextGetGraphDebugInfo(
        TFE_Context* ctx, const char* function_name, TF_Buffer* buf,
        TF_Status* status);
    
    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)
  5. tensorflow/c/eager/c_api_experimental.cc

                                   TF_Buffer* buf, TF_Status* status) {
      auto* function_def = tensorflow::unwrap(ctx)->FindFunctionDef(function_name);
      if (function_def == nullptr) {
        status->status = tensorflow::errors::NotFound(
            "Unable to find FunctionDef with name: ", function_name);
        return;
      }
      string str = function_def->SerializeAsString();
    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)
  6. tensorflow/c/eager/BUILD

            "@com_google_absl//absl/strings",
        ],
    )
    
    cc_library(
        name = "graph_function",
        srcs = ["graph_function.cc"],
        hdrs = [
            "graph_function.h",
        ],
        visibility = [
            "//tensorflow:internal",
        ],
        deps = [
            ":abstract_function",
            "//tensorflow/core:framework",
            "//tensorflow/core/platform:refcount",
    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)
  7. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      string op_type_;
      const char* op_name_ = nullptr;
      // TODO(srbs): Use this.
      string device_name_;
    };
    
    // GraphContext wraps a TF_Graph modeling a single function and manages the
    // "execution" of operation, i.e. adding them to the function.
    class GraphContext : public TracingContext {
     public:
      explicit GraphContext(const char* name)
          : TracingContext(kGraph),
            graph_(new TF_Graph(), TF_DeleteGraph),
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  8. tensorflow/c/eager/dlpack.cc

        default:
          return tensorflow::errors::InvalidArgument("Unsupported Type Codes: ",
                                                     dtype.code);
      }
    }
    
    // Wraps the deleter function of DLManagedTensor to match the function signature
    // TFE_NewTensorHandleFromDeviceMemory.
    void DeallocatorWrapperFunc(void* data, size_t len, void* dlmt_vptr) {
      TFE_CallDLManagedTensorDeleter(dlmt_vptr);
    }
    
    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)
  9. tensorflow/c/eager/tape.h

      // (`output_tensors`).
      //
      // If provided, a non-null `forward_function` will be used instead of the
      // backward function (`backward_function_getter` /
      // `backward_function_deleter`) to compute jvps for this operation. If
      // `forward_function` is null, a GradientTape is used on the backward function
      // to compute the jvp, which will waste computation when executing eagerly.
      //
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  10. CONTRIBUTING.md

        for details.
    
    #### Running doctest for testable docstring
    
    There are two ways to test the code in the docstring locally:
    
    1.  If you are only changing the docstring of a class/function/method, then you
        can test it by passing that file's path to
        [tf_doctest.py](https://www.tensorflow.org/code/tensorflow/tools/docs/tf_doctest.py).
        For example:
    
        ```bash
    Plain Text
    - Registered: Tue May 07 12:40:20 GMT 2024
    - Last Modified: Thu Mar 21 11:45:51 GMT 2024
    - 15.6K bytes
    - Viewed (0)
Back to top