- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 37 for add_input (0.13 sec)
-
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/c/eager/gradients.cc
forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) { TF_RETURN_IF_ERROR(op_->AddInput(input)); forward_op_->inputs.push_back(input); return absl::OkStatus(); } Status AddInputList(AbstractOperation* op_,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
return parent_op_->DeviceName(); } Status TapeOperation::SetDeviceName(const char* name) { return parent_op_->SetDeviceName(name); } Status TapeOperation::AddInput(AbstractTensorHandle* input) { TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return absl::OkStatus(); } Status TapeOperation::AddInputList( absl::Span<AbstractTensorHandle* const> inputs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 28 06:16:45 UTC 2024 - 9K 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/c/eager/unified_api_testutil.cc
} AbstractOperationPtr fn_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); for (auto input : inputs) { TF_RETURN_IF_ERROR(fn_op->AddInput(input)); } int retvals = outputs.size() - null_indices.size(); std::vector<AbstractTensorHandle*> fn_outputs(retvals); TF_RETURN_IF_ERROR(fn_op->Execute(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 13:57:45 UTC 2024 - 5.7K 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/tfr/integration/node_expansion_pass.cc
auto* new_op = new EagerOperation(&ctx); TF_RETURN_IF_ERROR( new_op->Reset(fname.c_str(), orig_op->DeviceName().c_str())); for (auto input : orig_op->GetInputs()) { TF_RETURN_IF_ERROR(new_op->AddInput(input)); } new_op->MutableAttrs()->CopyAttributes(orig_op->Attrs()); out_op->reset(new_op); LOG_FIRST_N(INFO, 1) << "Finish Node Expansion Passes. Rewrite the op to call function: " << fname;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Feb 25 16:22:36 UTC 2024 - 3.8K 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)