Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for FunctionBody (0.14 sec)

  1. tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc

      for (const std::string& func_name :
           function_library->ReachableDefinitions(graph_def).ListFunctionNames()) {
        const FunctionDef* func_def = function_library->Find(func_name);
        std::unique_ptr<FunctionBody> func_body;
        absl::Status status = FunctionDefToBodyHelper(
            *func_def, AttrSlice(&func_def->attr()), function_library, &func_body);
        // This is not expected to happen in practice
        if (!status.ok()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 12:22:33 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/force_xla_constants_on_host_pass.cc

      FunctionLibraryRuntime* flr =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
    
      for (Node* node : graph->nodes()) {
        if (CanCreateXlaKernel(node->def())) {
          const FunctionBody* fbody = nullptr;
          std::vector<int> constant_arg_indices;
          std::vector<int> resource_arg_indices;
    
          NameAttrList function;
          TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node->def(), &function));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 2.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/xla_kernel_creator.cc

      XlaOpRegistry::RegisterCompilationKernels();
    
      // Get function body, constant args, and resource args.
      NameAttrList function;
      TF_RETURN_IF_ERROR(NameAndAttrsFromFunctionCall(node_def, &function));
      const FunctionBody* fbody = nullptr;
      std::vector<int> constant_arg_indices;
      std::vector<int> resource_arg_indices;
      TF_RETURN_IF_ERROR(GetBodyAndConstantsAndResources(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 22:24:01 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/import_model.h

    // [Experimental]
    // Given a Function, returns a MLIR module containing the graph, expressed with
    // tf_executor dialect.
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ConvertFunctionToMlir(
        const FunctionBody* fbody, const FunctionLibraryDefinition& flib_def,
        mlir::MLIRContext* context);
    
    // Given a SavedModel, returns a MLIR module containing the functions, expressed
    // with tf_executor dialect.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 6.8K bytes
    - Viewed (0)
Back to top