- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for mutable_attr (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/upgrade_graph.cc
(*node_def.mutable_attr())["shared_name"].set_s(node_def.name()); } else { // Use the concat of function name and node name for such ops in a // function as the shared_name. "@" is used as the separator because // it is not allowed in the function name or the node name. (*node_def.mutable_attr())["shared_name"].set_s(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 14:33:47 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/force_xla_constants_on_host_pass_test.cc
}}}, {{"c", "t0:y:0"}}); AttrValue true_attribute; true_attribute.set_b(true); (*called_func.mutable_attr())[kXlaMustCompileAttr] = true_attribute; *library.add_function() = called_func; TF_ASSERT_OK(root.graph()->AddFunctionLibrary(library)); FunctionLibraryDefinition flib_def(OpRegistry::Global(), library);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
attrs) { NodeDef node; node.set_name(std::string(name)); node.set_op(std::string(op)); for (const auto& input : inputs) node.add_input(input); for (const auto& attr : attrs) node.mutable_attr()->insert({attr.first, attr.second.proto}); return node; } } // namespace GraphDef DeviceCompilerSerializeTest::GetTestGraph( const PartialTensorShape& input_shape) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 08:24:16 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/jit/device_compilation_cluster_signature_test.cc
} void BM_BuildSignature(::testing::benchmark::State& state) { const int n_args = state.range(0); NameAttrList fn; fn.set_name("afunction"); for (int i = 0; i < n_args; i++) { (*fn.mutable_attr())[absl::StrCat("T", i)].set_type(DT_FLOAT); } std::vector<XlaCompiler::Argument> args(n_args); for (int i = 0; i < n_args; i++) { args[i].kind = (((i % 3) == 0) ? XlaCompiler::Argument::kConstant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.cc
LOG_FIRST_N(INFO, 1) << "Run Node Expansion Passes"; // Get the FunctionDef and insert that into the context const NodeDef& ndef = orig_op->MutableAttrs()->BuildNodeDef(); auto& ctx = orig_op->EagerContext(); Fprint128 cache_key = orig_op->MutableAttrs()->CacheKey(orig_op->DeviceName()); // Include soft placement policy in cache key since the placement strategy // can change and thus affect which kernel is picked.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K bytes - Viewed (0)