Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for FunctionBody (0.4 sec)

  1. tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc

      std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
      TF_CHECK_OK(s.ToGraph(g.get()));
    
      std::vector<std::unique_ptr<FunctionBody>> fbodies;
      TF_CHECK_OK(RearrangeFunctionArguments(
          [&](const NameAttrList &function, const FunctionBody **fbody) {
            std::unique_ptr<FunctionBody> new_fbody;
            TF_RETURN_IF_ERROR(FunctionDefToBodyHelper(*fld.Find(function.name()),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 11:36:41 UTC 2024
    - 10.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc

          host_compute_core, &fld, &shape_inference_graphs,
          &has_outside_compilation));
    
      // Get rewritten XLA computation function.
      std::unique_ptr<FunctionBody> xla_fbody;
      TF_CHECK_OK(FunctionDefToBodyHelper(*fld.Find("cluster_rewritten"),
                                          AttrSlice(), &fld, &xla_fbody));
      auto node_name_index = xla_fbody->graph->BuildNodeNameIndex();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 41K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/compilability_check_util.h

    Status GetBodyAndConstantsAndResources(FunctionLibraryRuntime* flr,
                                           const NameAttrList& function,
                                           const FunctionBody** fbody,
                                           std::vector<int>* constant_arg_indices,
                                           std::vector<int>* resource_arg_indices);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 14.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/compilability_check_util.cc

                << uncompilable_reason << " : " << s;
        return false;
      }
    
      auto release_handle_on_return = gtl::MakeCleanup(
          [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); });
      const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle);
      bool is_compilable = true;
      for (const Node* node : fbody->graph->op_nodes()) {
        stack_trace->emplace_back(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/get_compiler_ir.cc

          TF_RETURN_IF_ERROR(graph.status());
          return *graph;
        }
      }
    }
    
    static absl::StatusOr<std::vector<XlaCompiler::Argument>>
    BuildXlaCompilerArgumentFromTensorSpec(
        const FunctionBody* fbody, absl::Span<int const> must_be_constant_idxs,
        absl::Span<const Tensor* const> inputs,
        absl::Span<VariableInfo const> variable_args,
        absl::Span<const ArgShapeAndDType> flat_arg_shape_and_dtype) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 06:59:07 UTC 2024
    - 19K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/python/mlir.cc

      if (fdef == nullptr) {
        s = tensorflow::errors::NotFound("Cannot find function ", function_name);
        tsl::Set_TF_Status_from_Status(status, s);
        return "// error";
      }
    
      std::unique_ptr<tensorflow::FunctionBody> fbody;
      s = FunctionDefToBodyHelper(*fdef, tensorflow::AttrSlice(), &flib_def,
                                  &fbody);
      if (!s.ok()) {
        tsl::Set_TF_Status_from_Status(status, s);
        return "// error";
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/xla_cluster_util.cc

          return true;
        }
    
        auto release_handle_on_return = gtl::MakeCleanup(
            [&] { TF_CHECK_OK(lib_runtime->ReleaseHandle(handle)); });
    
        const FunctionBody* fbody = lib_runtime->GetFunctionBody(handle);
        TF_RETURN_IF_ERROR(GetNodesRelatedToRefVariablesInDirection(
            *fbody->graph, lib_runtime, direction, depth + 1, &callee_ref_nodes));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
Back to top