Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 37 for function (0.21 sec)

  1. tensorflow/c/eager/graph_function.cc

    limitations under the License.
    ==============================================================================*/
    #include "tensorflow/c/eager/graph_function.h"
    
    #include <utility>
    
    #include "tensorflow/c/eager/abstract_function.h"
    #include "tensorflow/core/framework/function.h"
    #include "tensorflow/core/platform/status.h"
    
    namespace tensorflow {
    namespace tracing {
    namespace graph {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Mar 04 19:49:06 GMT 2024
    - 1.3K bytes
    - Viewed (0)
  2. 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)
  3. tensorflow/c/c_api_function_test.cc

      EXPECT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(s_));
      EXPECT_EQ(string("TF_Output scalar:0 is neither in the function body nor "
                       "among function inputs. Encountered while creating "
                       "function 'MyFunc'"),
                string(TF_Message(s_)));
    }
    
    void DefineFunction(const char* name, TF_Function** func,
                        const char* description = nullptr,
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
  4. tensorflow/c/eager/c_api_unified_experimental.cc

    using tensorflow::tracing::TracingTensorHandle;
    
    void TF_SetTracingImplementation(const char* name, TF_Status* s) {
      tsl::Set_TF_Status_from_Status(s, SetDefaultTracingEngine(name));
    }
    
    // Creates a new TensorFlow function, it is an execution context attached to a
    // given tracing context.
    TF_ExecutionContext* TF_CreateFunction(const char* fn_name, TF_Status* s) {
      return wrap(CreateTracingExecutionContext(fn_name, s));
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 9K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

          parent(nullptr),
          parent_inputs(nullptr) {
      // Tell the shape refiner to also run shape inference on functions.
      refiner.set_function_library_for_shape_inference(&graph.flib_def());
    }
    
    TF_Graph* TF_NewGraph() { return new TF_Graph; }
    
    void TF_DeleteGraph(TF_Graph* g) {
      if (g == nullptr) return;
      g->mu.lock();
      g->delete_requested = true;
      const bool del = g->sessions.empty();
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/experimental/gradients/grad_test_helper.cc

        absl::Span<AbstractTensorHandle* const> inputs, bool use_function,
        double abs_error) {
      auto num_inputs = inputs.size();
      std::vector<AbstractTensorHandle*> outputs(num_inputs);
      auto s = RunModel(grad_model, ctx, inputs, absl::MakeSpan(outputs),
                        /*use_function=*/use_function);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    
      for (int i = 0; i < num_inputs; ++i) {
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  7. 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)
  8. tensorflow/c/experimental/gradients/custom_gradient_test.cc

      AbstractTensorHandle* exp_output;
      TF_RETURN_IF_ERROR(ops::Exp(ctx, inputs[0], &exp_output, "Exp"));
      std::unique_ptr<GradientFunction> gradient_function(
          new PassThroughGradientFunction);
      tape.RecordOperation(inputs, {exp_output}, gradient_function.release());
      TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx,
                                              /*targets*/ {exp_output},
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 4.8K bytes
    - Viewed (0)
  9. 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)
  10. 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)
Back to top