- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for AddNodeAttr (0.12 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) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); AddNodeAttr("shapes", std::vector<TensorShapeProto>{}, node_def); } AddNodeAttr("ancestors", std::vector<string>{}, node_def); AddNodeAttr("Tinputs", recv_at_host_dtypes, node_def); AddNodeAttr("Toutputs", send_from_host_dtypes, node_def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
// outputs into host-memory constants and device-memory non-constants. AddNodeAttr(kXlaCompiledKernelAttr, true, node); AddNodeAttr(kXlaNumConstantArgsAttr, num_consts, node); AddNodeAttr(kXlaNumResourceArgsAttr, num_resources, node); return absl::OkStatus(); }; TF_RETURN_WITH_CONTEXT_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
} // Update op name, drop inputs and set attributes required by the Placeholder // op. *node->mutable_op() = "Placeholder"; node->clear_attr(); node->clear_input(); AddNodeAttr("dtype", dtype, node); AddNodeAttr("shape", it->second.shape, node); return absl::OkStatus(); } // Preprocesses GraphDef before it can be converted to Graph by,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)