- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 51 for node_def (0.53 sec)
-
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/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) -
tensorflow/compiler/mlir/lite/transforms/lift_tflite_flex_ops.cc
op_state.addTypes(op.getResultTypes()); SmallVector<NamedAttribute, 2> attrs; std::string parsed_op_name; tensorflow::NodeDef node_def; if (failed(ParseCustomOption(op.getCustomOption().getValue(), op.getLoc(), parsed_op_name, attrs, node_def))) { return failure(); } if (parsed_op_name != tf_op_name) { return op.emitOpError(
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/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/compiler/mlir/tensorflow/transforms/constant_fold_utils.cc
absl::StatusOr<OpKernelRunner> runner = OpKernelRunner::Create( node_def->get()->op(), node_def->get()->name(), host_cpu, operands.size(), [&](tensorflow::AttrValueMap* attr_value_map) { *attr_value_map = node_def->get()->attr(); return absl::OkStatus(); }, fallback_state.device_manager(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py
tf_graph_def = root.signatures['serving_default'].graph.as_graph_def() count = 0 for node_def in tf_graph_def.node: if node_def.op == 'XlaCallModule': count += 1 for function in tf_graph_def.library.function: for node_def in function.node_def: if node_def.op == 'XlaCallModule': count += 1 return count def _get_function_aliases(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 18.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc
.Input(inputs) .Attr("ids", ids) .Attr("calibration_methods", calibration_methods) .Finalize(node_def())); ASSERT_THAT(InitOp(), StatusIs(tsl::error::INVALID_ARGUMENT, HasSubstr("NodeDef missing attr 'output_file_path'"))); } TEST_F(CalibrationStatisticsSaverTest, WrongNumInputs) { std::vector<std::string> ids{"1"};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
if (stateless && stateless.getValue()) *node_def->mutable_op() = "Stateless" + node_def->op(); } // Add inputs to the NodeDef based on the number of operands. This is required // as later when edges are added to the Node using Graph::AddEdge the // associated NodeDef is not updated. for (int i = 0, e = inst->getNumOperands(); i < e; ++i) { node_def->add_input(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 19.7K 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)