- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 21 for set_ip (0.27 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)