- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 185 for nodedef (0.1 sec)
-
tensorflow/compiler/jit/tests/device_compiler_test_helper.cc
tensor.flat<float>()(i) = offset + i; } return tensor; } NodeDef MakeNode( absl::string_view name, absl::string_view op, absl::Span<const std::string> inputs, absl::Span< const std::pair<std::string, FunctionDefHelper::AttrValueWrapper>> attrs) { NodeDef node; node.set_name(std::string(name)); node.set_op(std::string(op));
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/mlir/quantization/stablehlo/cc/saved_model_export_test.cc
namespace mlir::quant::stablehlo { namespace { using ::tensorflow::AssetFileDef; using ::tensorflow::GraphDef; using ::tensorflow::NodeDef; using ::tensorflow::SaverDef; using ::tensorflow::quantization::ExportedModel; using ::testing::HasSubstr; using ::testing::IsEmpty; using ::testing::SizeIs; using ::testing::StrEq; using ::tsl::protobuf::TextFormat; using ::tsl::testing::IsOk; using ::tsl::testing::StatusIs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
absl::StatusOr<std::unique_ptr<tensorflow::NodeDef>> node_def = tensorflow::ConvertTFDialectOpToNodeDef( op, /*name=*/"", /*ignore_unregistered_attrs=*/true); if (!node_def.ok()) { op->emitError("Failed to obtain TensorFlow NodeDef: " + node_def.status().ToString()); return false; } std::string node_def_str; if (!(*node_def)->SerializeToString(&node_def_str)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/node_expansion_pass.h
Status Run(EagerOperation* orig_op, std::unique_ptr<tensorflow::EagerOperation>* out_op) override; private: // Whether to run this pass. If this is enabled, the NodeDef will be imported // to MLIR even no tf composition file is found. bool IsEnabled() { const char* tfr_lib_env_val = getenv(string(kTFRLibEnv).c_str()); return tfr_lib_env_val != nullptr; } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
const FunctionDef *true_def = fld.Find("oc_then_branch_host_if_true_fn"); EXPECT_NE(true_def, nullptr); bool has_identity_true_fn_node = false; for (const auto &node_def : true_def->node_def()) { if (node_def.name() == "identity_true_fn") { has_identity_true_fn_node = true; break; } } EXPECT_TRUE(has_identity_true_fn_node);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/jit/compilability_check_util.h
std::vector<int>* resource_arg_indices); // Given a NodeDef `node_def` returns true iff `node_def` has kXlaCompileAttr // set. bool CanCreateXlaKernel(const NodeDef& node_def); // Returns memory types for the input. // `constant_arg_indices` and `resource_arg_indices` are sorted arrays of
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
// runtime OpKernel. void RemoveIdentityCast(NodeDef* node_def) { auto attr = node_def->mutable_attr(); if (node_def->op() == "Cast" && attr->contains("SrcT") && 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")}});
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/cc/experimental/libexport/load.h
// Since we may need to load many constants, we create a hash map of these // names to their corresponding nodes at load time in order to look them up // in constant time. tensorflow::StatusOr<const tensorflow::NodeDef*> GetGraphDefNode( std::string name); // Returns a list of function defs in the SavedModel. const protobuf::RepeatedPtrField<FunctionDef>& GetFunctionDefs();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jul 13 06:33:42 UTC 2023 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_compile_util.h
kUncompiled, kCompiling, kCompiled, }; // Creates a single-node graph using the specified `node_def` as the only op // apart from the arg and retval nodes corresponding to `args` and // `result_types` respectively. absl::StatusOr<std::unique_ptr<Graph>> CreateSingleOpGraph( const NodeDef& node_def, absl::Span<const XlaArgument> args, absl::Span<const DataType> result_types);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
absl::StatusOr<FunctionDef> TFRDecomposeContext::ExpandNode( const NodeDef& node_def, StringPiece func_name) { const OpDef* op_def; TF_RETURN_IF_ERROR(OpRegistry::Global()->LookUpOpDef(node_def.op(), &op_def)); DataTypeVector input_dtys, output_dtys; TF_RETURN_IF_ERROR(InputTypesForNode(node_def, *op_def, &input_dtys)); TF_RETURN_IF_ERROR(OutputTypesForNode(node_def, *op_def, &output_dtys));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0)