- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 76 for nodedef (0.17 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def.h
void MutateNodeDefs(tensorflow::GraphDef& graph_def, FuncT&& func) { for (tensorflow::NodeDef& node_def : *graph_def.mutable_node()) { func(node_def); } for (tensorflow::FunctionDef& function_def : *graph_def.mutable_library()->mutable_function()) { for (tensorflow::NodeDef& node_def : *function_def.mutable_node_def()) { func(node_def); } } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/graph_def_test.cc
MutateNodeDefs(graph_def, [](NodeDef& node_def) { node_def.set_name("bar"); }); ASSERT_THAT(graph_def.library().function(), SizeIs(1)); ASSERT_THAT(graph_def.library().function()[0].node_def(), SizeIs(1)); EXPECT_THAT(graph_def.library().function()[0].node_def()[0].name(), StrEq("bar")); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 28 18:38:06 UTC 2023 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.h
StringPiece tfr_raw_text, mlir::MLIRContext* mlir_ctx); // Decomposes the op in the NodeDef to a set of primitive ops according to the // decompose library in the context. Wrap the decomposed result in a // FunctionDef. absl::StatusOr<FunctionDef> ExpandNode(const NodeDef& node_def, StringPiece func_name); // Runs the decompose passes on the user_module.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 11:12:54 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/c/kernels/bitcast_op_test.cc
return cpu_allocator(); } }; void TestBitcastOp(Tensor* input_tensor, DataType out_type, TensorShape expected_shape, error::Code expected_code) { Status status; NodeDef def; def.set_op("Bitcast"); def.set_device(DEVICE_CPU); AttrValue typeAttr; SetAttrValue(input_tensor->dtype(), &typeAttr); AttrValue outTypeAttr; SetAttrValue(out_type, &outTypeAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jul 18 15:10:51 UTC 2022 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_kernel_creator.h
#include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/node_properties.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { class FunctionLibraryRuntime; class OpKernel; class XlaKernelCreator : public CustomKernelCreator { public: // Given a NodeDef 'node_def' and the function library runtime 'flr', returns
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 26 19:43:17 UTC 2023 - 1.8K bytes - Viewed (0) -
tensorflow/c/c_test_util.h
bool IsPlaceholder(const tensorflow::NodeDef& node_def); bool IsScalarConst(const tensorflow::NodeDef& node_def, int v); bool IsAddN(const tensorflow::NodeDef& node_def, int n); bool IsNeg(const tensorflow::NodeDef& node_def, const string& input); bool GetGraphDef(TF_Graph* graph, tensorflow::GraphDef* graph_def); bool GetNodeDef(TF_Operation* oper, tensorflow::NodeDef* node_def);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 09 01:06:53 UTC 2018 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx_test.cc
std::vector<NodeAndType> nodes; for (auto& node : graph.node_def()) { nodes.push_back({node.op(), node.attr().at("T").type()}); } return nodes; } TEST_F(TFRDecomposeContextTest, FLOAT_1_ins) { std::vector<NodeDefBuilder::NodeOut> src_list; src_list.emplace_back("input", 0, DT_FLOAT); NodeDef test_node; auto status = NodeDefBuilder("float_add", "MyAddN")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 5.8K bytes - Viewed (0) -
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/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/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)