Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 21 for function (0.15 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/c_api_experimental.cc

    }
    
    char* TF_FunctionDebugString(TF_Function* func, size_t* len) {
      const auto& debug_str = DebugString(func->record->fdef());
      *len = debug_str.size();
      char* ret = static_cast<char*>(malloc(*len + 1));
      memcpy(ret, debug_str.c_str(), *len + 1);
      return ret;
    }
    
    // On success, returns a set of TF_Function instances from `text_proto` of
    // GraphDef type. These functions must be deleted by calling TF_DeleteFunction.
    //
    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)
  3. tensorflow/c/experimental/gradients/math_grad_test.cc

                           /*tfrt*/ ::testing::Values(false),
                           /*use_function*/ ::testing::Values(true, false)));
    #else
    INSTANTIATE_TEST_SUITE_P(
        UnifiedCAPI, CppGradients,
        ::testing::Combine(::testing::Values("graphdef", "mlir"),
                           /*tfrt*/ ::testing::Values(false),
                           /*use_function*/ ::testing::Values(true, false)));
    #endif
    }  // namespace
    }  // namespace internal
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Thu Apr 13 17:32:14 GMT 2023
    - 16.3K bytes
    - Viewed (0)
  4. tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc

    #include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h"
    #include "tensorflow/c/eager/tfe_context_internal.h"
    #include "tensorflow/core/common_runtime/eager/context.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/lib/core/status_test_util.h"
    #include "tensorflow/core/platform/test.h"
    
    namespace tensorflow {
    namespace parallel_device {
    
    using ::testing::ElementsAre;
    using ::testing::HasSubstr;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 15.3K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/c/c_test_util.cc

        grads.emplace_back(grad.function_name(), grad.gradient_func());
      }
      std::sort(grads.begin(), grads.end());
      return grads;
    }
    
    std::vector<string> GetFuncNames(const tensorflow::GraphDef& graph_def) {
      std::vector<string> names;
      auto functions = graph_def.library().function();
      names.reserve(functions.size());
      for (const tensorflow::FunctionDef& func : functions) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  7. 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)
  8. tensorflow/c/eager/c_api_distributed_test.cc

    // once (i.e., on the main function side) in running distributed functions.
    // This test creates a cluster with two workers, create a variable on the
    // second worker, and run a distributed function (VariableAddFunction) whose ops
    // span the local and remote workers. If the graph optimization pass is executed
    // on both the main function side and the component function side, an error will
    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)
  9. tensorflow/c/experimental/grappler/grappler.cc

    #include "tensorflow/c/tf_buffer.h"
    #include "tensorflow/c/tf_buffer_internal.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/c/tf_status_helper.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/framework/op.h"
    #include "tensorflow/core/framework/op_def.pb.h"
    #include "tensorflow/core/grappler/costs/graph_properties.h"
    #include "tensorflow/core/grappler/costs/op_performance_data.pb.h"
    C++
    - Registered: Tue Feb 27 12:39:08 GMT 2024
    - Last Modified: Wed Sep 06 19:12:29 GMT 2023
    - 15K bytes
    - Viewed (1)
  10. tensorflow/c/eager/parallel_device/parallel_device_test.cc

      TF_Operation* operations[]{placeholder_op, reduce_op};
      TF_Output y{reduce_op, 0};
      const char* output_name = "y";
      std::unique_ptr<TF_Function, decltype(&TF_DeleteFunction)> function(
          TF_GraphToFunction(
              /* fn_body */ body.get(), /* fn_name */ function_name,
              /* append_hash_to_fn_name */ 0, /* num_opers */ 2,
              /* opers */ operations, /* ninputs */ 1, /* inputs */ &x,
    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)
Back to top