Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 21 for set_ip (0.27 sec)

  1. tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc

      // may cause spurious cache misses.
      TF_ASSIGN_OR_RETURN(uint64 fingerprint, FingerprintGraph(*graph));
      VLOG(1) << "Subgraph fingerprint:" << fingerprint;
      call_def->set_op(absl::StrCat(call_def->op(), "_", fingerprint));
      return absl::OkStatus();
    }
    
    }  // namespace
    
    /*static*/ Status EncapsulateXlaComputationsPass::Encapsulate(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 15.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/build_xla_ops_pass_test.cc

                                 const string& node_name, int num_constant_args,
                                 int num_resource_args, Node** result) {
      NodeDef call_node;
      call_node.set_name(node_name);
      call_node.set_op(callee_name);
      AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node);
      AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node);
      AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 12.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/jit/compilability_check_util.cc

                                     NodeDef* node_def) {
      const NameAttrList* name_attr;
      TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), attr_name, &name_attr));
      node_def->set_op(name_attr->name());
      *(node_def->mutable_attr()) = name_attr->attr();
      return absl::OkStatus();
    }
    
    absl::StatusOr<std::vector<NodeDef>> MakeCallNodesFromAttribute(
        const Node& node, absl::string_view attr_name,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 30.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/jit/shape_inference.cc

            TensorShapeProto shape_proto;
            context->ShapeHandleToProto(handle, &shape_proto);
            if (!shape_proto.unknown_rank()) {
              NodeDef const_def;
              const_def.set_op("Const");
              Node* var_node;
              TF_RETURN_IF_ERROR(n->input_node(0, &var_node));
              const_def.set_name(
                  graph->NewName(absl::StrCat("var_shape_", var_node->name())));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 00:41:19 UTC 2024
    - 13K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc

          attr->contains("DstT") &&
          attr->at("SrcT").type() == attr->at("DstT").type() &&
          attr->contains("Truncate") && !attr->at("Truncate").b()) {
        node_def->set_op("Identity");
        attr->insert({{"T", attr->at("SrcT")}});
        attr->erase("SrcT");
        attr->erase("DstT");
        attr->erase("Truncate");
      }
    }
    
    }  // namespace
    
    Status GetAttrValuesFromOperation(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 11.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc

    Value LocalVarOp_resource(LocalVarOp &op) {
      if (auto var_handle_op = std::get_if<TF::VarHandleOp>(&op)) {
        return var_handle_op->getResource();
      } else {
        return std::get<TF::MlirLocalVarOp>(op).getResource();
      }
    }
    
    void LocalVarOp_erase(LocalVarOp &op) {
      if (auto var_handle_op = std::get_if<TF::VarHandleOp>(&op)) {
        var_handle_op->erase();
      } else {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/node_matchers.cc

      props.set_name(std::move(name));
      return props;
    }
    
    // Matches a node with op `op`.
    impl::NodeMatcherProperties Op(string op) {
      impl::NodeMatcherProperties props;
      props.set_op(std::move(op));
      return props;
    }
    
    // Matches a node with assigned device `assigned_device`.
    impl::NodeMatcherProperties AssignedDevice(string assigned_device) {
      impl::NodeMatcherProperties props;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 03 16:15:20 UTC 2022
    - 16.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/jit/resource_operation_safety_analysis_test.cc

      return flib_def;
    }
    
    Node* MakeCall(Graph* graph, const string& callee_name, const string& node_name,
                   Status* status) {
      NodeDef call_node;
      call_node.set_name(node_name);
      call_node.set_op(callee_name);
      return graph->AddNode(call_node, status);
    }
    
    TEST(ResourceOperationSafetyAnalysisTest, CallRead) {
      Scope root = Scope::NewRootScope().ExitOnError();
    
      FunctionDefLibrary flib_def =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 28 16:53:59 UTC 2020
    - 18.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/jit/xla_cluster_util.cc

        const absl::flat_hash_map<absl::string_view, int>& histogram) {
      for (const auto& pair : histogram) {
        XlaAutoClusteringSummary::OpAndCount* new_entry = result->Add();
        new_entry->set_op(std::string(pair.first));
        new_entry->set_count(pair.second);
      }
    
      absl::c_sort(*result, [](const XlaAutoClusteringSummary::OpAndCount& a,
                               const XlaAutoClusteringSummary::OpAndCount& b) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  10. tensorflow/c/c_api_experimental.cc

      const int num_inputs = input_shapes->num_items;
      NodeDef node_def;
      tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op);
      node_def.set_name(op->Name());
      node_def.set_op(op->Name());
      for (int i = 0; i < num_inputs; ++i) {
        node_def.add_input("dummy_input");
      }
      OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 29.4K bytes
    - Viewed (0)
Back to top