- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for add_input (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
// 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(); } if (auto attr = inst->getAttrOfType<mlir::StringAttr>("device")) { node_def->set_device(std::string(attr.getValue())); } // Add the node debug info. TF_RETURN_IF_ERROR(ConvertLocation(
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/c/c_api_experimental.cc
NodeDef node_def; tensorflow::ImmediateExecutionOperation* op = tensorflow::unwrap(tfe_op); node_def.set_name(op->Name()); node_def.set_op(op->Name()); for (int i = 0; i < num_inputs; ++i) { node_def.add_input("dummy_input"); } OperationFromInterface(op)->Attrs().FillAttrValueMap(node_def.mutable_attr()); const tensorflow::OpRegistrationData* op_reg_data; status->status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/core/ops/variable_ops.cc
TF_RETURN_IF_ERROR(assign_op->Reset("AssignVariableOp", nullptr)); TF_RETURN_IF_ERROR(assign_op->SetAttrType("dtype", dtype)); TF_RETURN_IF_ERROR(assign_op->AddInput(variable_handle)); TF_RETURN_IF_ERROR(assign_op->AddInput(value)); int num_retvals = 0; TF_RETURN_IF_ERROR(assign_op->Execute({}, &num_retvals)); return Status(); } Status ReadVariable(ImmediateExecutionContext* ctx,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 11:28:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool_patterns.td
(MHLO_CompositeOp:$old_val (variadic $a_input), ConstantStrAttr<StrAttr, "aten.avg_pool2d.default">, $attrs, $_, $_), (TFL_TransposeOp (TFL_AveragePool2DOp /*input*/ (TFL_TransposeOp $a_input, (Arith_ConstantOp
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/cpp/views/arg_view.cc
ArgView::ArgView(ArgSpec arg) : arg_(arg) {} string ArgView::VariableName() const { return arg_.name(); } string ArgView::SetterMethod() const { if (IsList()) { return "AddInputList"; } else { return "AddInput"; } } std::vector<string> ArgView::SetterArgs() const { return {VariableName()}; } bool ArgView::IsList() const { return arg_.arg_type().is_list(); } int ArgView::Position() const { return arg_.position(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 07:02:00 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize_xla.mlir
%bias = arith.constant dense<[7.11401462, 7.05456924]> : tensor<2xf32> %q_input= "quantfork.qcast"(%input) : (tensor<1x3x4x3xf32>) -> tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>> %dq_input= "quantfork.dcast"(%q_input) : (tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>>) -> tensor<1x3x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize_layout.cc
LogicalResult matchAndRewrite(stablehlo::PadOp pad_op, PatternRewriter& rewriter) const override { Value pad_input = pad_op.getOperand(); RankedTensorType pad_type = pad_op.getType().cast<RankedTensorType>(); auto transpose_op = pad_input.getDefiningOp<stablehlo::TransposeOp>(); if (!transpose_op || !transpose_op->hasOneUse()) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
SmallVector<Value, 8> operands_; llvm::StringMap<Attribute> attrs_; std::unique_ptr<OperationState> state_; // This is the index of the next ODS operand that will be added with AddInput // or AddInput; int current_ods_input_ = 0; const tensorflow::OpDef* op_def_ = nullptr; const char* op_name_ = nullptr; string tf_op_type_; // TODO(srbs): Use this. string device_name_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/quantize.mlir
%bias = arith.constant dense<[7.11401462, 7.05456924]> : tensor<2xf32> %q_input= "quantfork.qcast"(%input) : (tensor<1x3x4x3xf32>) -> tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>> %dq_input= "quantfork.dcast"(%q_input) : (tensor<1x3x4x3x!quant.uniform<i8:f32, 0.58810077742034317:-128>>) -> tensor<1x3x4x3xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:32:28 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api_unified_experimental.cc
TF_AbstractTensor* const* inputs, TF_OutputList* o, TF_Status* s) { for (int i = 0; i < num_inputs; i++) { tsl::Set_TF_Status_from_Status(s, unwrap(op)->AddInput(unwrap(inputs[i]))); if (TF_GetCode(s) != TF_OK) { return; } } int num_outputs = unwrap(o)->expected_num_outputs; tsl::Set_TF_Status_from_Status( s, unwrap(op)->Execute(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 10:15:17 UTC 2024 - 9K bytes - Viewed (0)