Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for rtype_String (0.17 sec)

  1. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

        const OpRegistrationData* op_reg_data;
        TF_RETURN_IF_ERROR(library->LookUp(node->type_string(), &op_reg_data));
        if (op_reg_data->is_function_op) {
          function_nodes.push_back(node);
        }
      }
      for (auto node : function_nodes) {
        VLOG(2) << "Inlining function " << node->name();
        const FunctionDef* fdef = library->Find(node->type_string());
        if (fdef == nullptr) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/mark_for_compilation_pass.cc

        return n->type_string() == "NoOp" && n->out_edges().size() == 1 &&
               (*n->out_edges().begin())->dst()->IsSink();
      }
    
      return false;
    }
    
    bool MarkForCompilationPassImpl::IsScalarIntegerResourceOperation(
        const Cluster& cluster) {
      Node* n = GetOnlyNodeIn(cluster);
      if (!n) {
        return false;
      }
    
      if (n->type_string() != "AssignAddVariableOp" &&
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 12:19:41 UTC 2024
    - 85.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/deadness_analysis.cc

        bool is_boolean_tensor =
            BaseType(node->output_type(tensor_id.index())) == DT_BOOL;
        TF_RET_CHECK(!must_be_true || is_boolean_tensor);
    
        if (node->type_string() == "Const" && must_be_true) {
          const TensorProto* proto = nullptr;
          TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "value", &proto));
    
          Tensor tensor(proto->dtype());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 60.4K bytes
    - Viewed (0)
Back to top