- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 185 for nodedef (0.1 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
// ShapedType for the leading values with ShapedType in the results of the // nodes. Set it to true if the returned NodeDef will be executed by the linked // TF Eager runtime. absl::StatusOr<std::unique_ptr<NodeDef>> ConvertTFDialectOpToNodeDef( mlir::Operation* inst, llvm::StringRef name, bool ignore_unregistered_attrs); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
StringRef custom_options, Location loc, std::string& op_name, SmallVectorImpl<NamedAttribute>& attributes, tensorflow::NodeDef& node_def) { // The flexbuffer contains a vector where the first elements is the // op name and the second is a serialized NodeDef. const uint8_t* const opt_data = reinterpret_cast<const uint8_t*>(custom_options.data()); const size_t opt_size = custom_options.size();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.h
// 'input_permutation' and 'output_permutation' are initialized to the identity // permutation. 'nodedef' is the NodeDef for the call to the function under // construction, provided to allow additional attributes to be set. // The rewrite may also change the NodeDef's operator name, and that // name will be used as the name of the generated function. typedef std::function<Status(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 12 03:59:36 UTC 2022 - 4.9K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
} } } return found_dtype && found_value; } bool IsAddN(const tensorflow::NodeDef& node_def, int n) { if (node_def.op() != "AddN" || node_def.name() != "add" || node_def.input_size() != n) { return false; } bool found_t = false; bool found_n = false; for (const auto& attr : node_def.attr()) { if (attr.first == "T") { if (attr.second.type() == tensorflow::DT_INT32) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 15 03:16:52 UTC 2021 - 17.8K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
bool HasLiftedArgs(const FunctionDef& function_def) { return absl::c_any_of(function_def.node_def(), [](const NodeDef& node_def) { return (node_def.op() == "Placeholder" && node_def.attr().find(kXlaLiftedArgOutsideCompilationAttrName) != node_def.attr().end()); }); } // Find lifted arguments in a function body and their corresponding outside // compilation nodes.
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/mlir/tensorflow/tests/mlir2graphdef/infer_derived_attribute.mlir
func.func @main() { // The operation does not have any attributes, but TensorFlow OpDef expects // a `dtype` to be added on the NodeDef. We verify that we correctly use the // DerivedAttr to populate the NodeDef. // CHECK: key: "dtype" // CHECK-NEXT: value { // CHECK-NEXT: type: DT_FLOAT // CHECK: float_val: 2 // CHECK: key: "dtype" // CHECK-NEXT: value {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 1K bytes - Viewed (0) -
tensorflow/c/ops_test.cc
PartialTensorShape S(std::initializer_list<int64_t> dims) { return PartialTensorShape(dims); } PartialTensorShape Unknown() { return PartialTensorShape(); } TEST(OpsTest, ShapeInferenceWithRank) { NodeDef def; shape_inference::InferenceContext c(0, def, MakeOpDef(1, 0), {S({10, 20, 30})}, {}, {}, {}); shape_inference::ShapeHandle in0 = c.input(0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 11 01:20:50 UTC 2021 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.cc
node_def->mutable_attr())); return node_def; } absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetReturnNode( FuncOp function, Value operand, unsigned index, llvm::StringRef name) { auto node_def = std::make_unique<NodeDef>(); if (!name.empty()) node_def->set_name(std::string(ParseTensorName(name.str()).node())); else
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
node_def->mutable_attr())); return node_def; } absl::StatusOr<std::unique_ptr<NodeDef>> Exporter::GetReturnNode( FuncOp function, Value operand, unsigned index, llvm::StringRef name) { auto node_def = std::make_unique<NodeDef>(); if (!name.empty()) node_def->set_name(std::string(ParseTensorName(name.str()).node())); else
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/mlir/tensorflow/utils/export_utils.h
// dialect back into a TensorFlow valid op name. absl::StatusOr<llvm::StringRef> GetTensorFlowOpName(llvm::StringRef); // Converts an MLIR operation to TensorFlow NodeDef with given node name. This // name should be unique to the graph it is being inserted into. absl::StatusOr<std::unique_ptr<NodeDef>> GetOperationNodeDef( mlir::Operation* inst, llvm::StringRef name); // Converts MLIR attributes with values to their tensorflow equivalent.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.9K bytes - Viewed (0)