- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 51 for _Arg (0.04 sec)
-
tensorflow/compiler/jit/device_executable_persistor_test.cc
bool mul = false) { std::unique_ptr<Graph> graph(new Graph(OpRegistry::Global())); Scope scope = Scope::NewRootScope().ExitOnError(); auto a = ops::_Arg(scope.WithOpName("A"), DT_INT32, 0); auto b = ops::_Arg(scope.WithOpName("B"), DT_INT32, 1); if (mul) { auto c = ops::Mul(scope.WithOpName("C"), a, b); auto d = ops::_Retval(scope.WithOpName("D"), c, 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/mlir_bridge_pass_util.cc
bool HasPsWithResourceVariable(const Graph& graph) { // Check parameter serverjobs and resource variable arguments that are // on parameter servers. const std::string jobType = "ps"; const std::string nodeType = "_Arg"; const std::string attrKey = "T"; for (const Node* node : graph.nodes()) { if (node->type_string() == nodeType) { auto device_name = node->assigned_device_name(); DeviceNameUtils::ParsedName device;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 12:22:33 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
// tensor is found (when there are no variables to restore, it is expected that // the file prefix tensor does not exist). The file prefix tensor is found among // the "_Arg" nodes, as it is translated from the MLIR @main function's // argument. It also must have the attribute `tf_saved_model.index_path = // ["__tf_file_prefix"]`. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
static const char kImportModelDefaultGraphFuncName[] = "main"; // Please refer to the TFG dialect description for the list of used attributes. // Belows are the attributes in TFE. // TFE Arguments and Results (Got from "_Arg", // "_Retval", .etc) // NodeDef.device <-> "tf.device" // NodeDef.attr <-> "tf." // // TFE general operations // NodeDef.device <-> "device" //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
src/runtime/runtime-gdb.py
class GoroutinesCmd(gdb.Command): "List all goroutines." def __init__(self): gdb.Command.__init__(self, "info goroutines", gdb.COMMAND_STACK, gdb.COMPLETE_NONE) def invoke(self, _arg, _from_tty): # args = gdb.string_to_argv(arg) vp = gdb.lookup_type('void').pointer() for ptr in SliceValue(gdb.parse_and_eval("'runtime.allgs'")): if ptr['atomicstatus']['value'] == G_DEAD: continue
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 10 12:59:20 UTC 2023 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
return errors::InvalidArgument( "Invalid \"_handle_dtypes\" attribute value for _Arg node: ", shape_attr->DebugString()); } if (shape_attr->list().shape().empty()) { return errors::InvalidArgument( "Invalid \"_handle_shapes\" attribute value for _Arg node: ", shape_attr->DebugString()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
node_builder.Input(inputs); return opts.FinalizeBuilder(&node_builder); } Node* ArgOp(int index, DataType type, const GraphDefBuilder::Options& opts) { return ops::SourceOp("_Arg", opts.WithAttr("T", type).WithAttr("index", index)); } Node* RetOp(int index, ops::NodeOut a, const GraphDefBuilder::Options& opts) { if (opts.HaveError()) return nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
TF_RET_CHECK(tensor_id.index() == 0) << "input port designation not supported"; // Only assign user of argument the input name if the main graph did not // have its _Arg nodes lifted into the functions arguments. // Ensure name does not get reused. (void)exporter.op_to_name_.GetUniqueName(name); } } // Adds nodes for basic block (function) arguments.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
TF_RET_CHECK(tensor_id.index() == 0) << "input port designation not supported"; // Only assign user of argument the input name if the main graph did not // have its _Arg nodes lifted into the functions arguments. // Ensure name does not get reused. (void)exporter.op_to_name_.GetUniqueName(name); } } // Adds nodes for basic block (function) arguments.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"XlaSpmdShardToFullShape", "XlaSvd", "XlaVariadicReduce", "XlaVariadicReduceV2", "XlaVariadicSort", "XlaWhile", "Zeta", "_Arg", "_ArrayToList", "_ListToArray", "_Retval"}; return result; } } // namespace testing
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0)