- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for AddNodeAttr (0.22 sec)
-
tensorflow/compiler/jit/encapsulate_xla_computations_pass.cc
def.set_device(launch->requested_device()); AddNodeAttr("Tconstants", DataTypeVector{}, &def); AddNodeAttr("Targs", arg_types, &def); AddNodeAttr("Nresources", num_variables, &def); AddNodeAttr("Tresults", output_types, &def); NameAttrList function; function.set_name(xla_function_info.function_name); AddNodeAttr("function", function, &def); for (Node* node : nodes_to_remove) {
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/shape_inference.cc
} else { value.add_int64_val(dim.size()); } } AddNodeAttr("value", value, &const_def); for (auto const& attr : n->attrs()) { if (*attr.first.begin() == '_') { AddNodeAttr(attr.first, attr.second, &const_def); } } TF_ASSIGN_OR_RETURN(Node * const_node, graph->AddNode(const_def));
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/jit/build_xla_ops_pass_test.cc
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); TF_ASSIGN_OR_RETURN(*result, graph->AddNode(call_node)); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 12.2K bytes - Viewed (0)