Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for GetFunctionBody (0.16 sec)

  1. tensorflow/compiler/jit/compilability_check_util.cc

        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(
            StackFrameView{node->name(), function.name(), node->GetStackTrace()});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc

                          xla_literal->element_count());
          TF_ASSIGN_OR_RETURN(auto attrs,
                              tensorflow::ConvertTensor(tensor, &builder));
          builder.setInsertionPoint(
              &func_op.getFunctionBody().getBlocks().front().front());
          // Use mhlo.Constant when it is consumed by the lowering passes since they
          // can't lower tf.Const.
          Value cst;
          if (use_mhlo_const) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 01:03:21 UTC 2024
    - 35.8K bytes
    - Viewed (0)
  3. 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));
    
        // We could possibly use something cheaper than
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/extract_outside_compilation_pass.cc

      auto cleanup_handle = gtl::MakeCleanup([&]() {
        auto s = flr->ReleaseHandle(handle);
        if (!s.ok()) {
          ret_status.Update(s);
        }
      });
      const FunctionBody* fbody = flr->GetFunctionBody(handle);
    
      // Check if we have outside compilation nodes.
      *has_outside_compilation = false;
      for (Node* n : fbody->graph->nodes()) {
        if (HasNodeAttr(n->def(), outside_compilation_attr_name)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 104.7K bytes
    - Viewed (0)
Back to top