- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 94 for ret_attrs (0.26 sec)
-
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
StringAttr::get(func_.getContext(), kCustomSSDPostprocessing)); OpBuilder builder(func_.getBody()); std::string custom_option_buffer; if (failed(CreateNMSCustomOptions(func_, attr_.getAttrs(), custom_option_buffer))) { return failure(); } auto op = builder.create<CustomOp>( func_.getLoc(), func_.getFunctionType().getResults(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
} } OpBuilder builder(op); builder.setInsertionPointAfter(op); OperationState state(op->getLoc(), op->getName().getStringRef(), op->getOperands(), result_types, op->getAttrs()); for (int i = 0; i < op->getNumRegions(); ++i) { state.addRegion(); } Operation* new_op = builder.create(state); for (const auto& indexed_regions : llvm::enumerate(op->getRegions())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
pkg/registry/batch/job/strategy.go
"status.successful": strconv.Itoa(int(job.Status.Succeeded)), } return generic.MergeFieldsSets(objectMetaFieldsSet, specificFieldsSet) } // GetAttrs returns labels and fields of a given object for filtering purposes. func GetAttrs(obj runtime.Object) (labels.Set, fields.Set, error) { job, ok := obj.(*batch.Job) if !ok { return nil, nil, fmt.Errorf("given object is not a job.") }
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Fri Mar 08 16:43:24 UTC 2024 - 18.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
} }); return entry_funcs; } LogicalResult GetCallees(SymbolUserOpInterface op, SymbolTable &symtab, llvm::SmallVector<func::FuncOp> &callees) { for (auto attr : op->getAttrs()) { auto sym = mlir::dyn_cast<SymbolRefAttr>(attr.getValue()); if (!sym) continue; auto callee = symtab.lookup<func::FuncOp>(sym.getRootReference()); if (!callee) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc
return func_.emitWarning() << "Invalid number of results from " << kMaxUnpooling << ": " << func_.getFunctionType().getNumResults(); } auto attrs = attr_.getAttrs(); if (failed(HasIntegerArrayWithSize(&func_, attrs, "pool_size", 2))) { return failure(); } if (failed(HasIntegerArrayWithSize(&func_, attrs, "strides", 2))) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
LogicalResult SmuggleOp(Operation* op, PatternRewriter& rewriter) { auto call_target = rewriter.getNamedAttr("call_target_name", op->getName().getIdentifier()); SmallVector<NamedAttribute> attrs{op->getAttrs()}; attrs.push_back(call_target); auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), attrs); rewriter.replaceOp(op, custom_call.getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
auto *new_ewise_op = rewriter.create(op->getLoc(), op->getName().getIdentifier(), {tpose_arg1.getOperand(0), tpose_arg2.getOperand(0)}, new_out_type, op->getAttrs()); // Apply original tranpose to output of ewise op. auto out_tpose_op = rewriter.create<TFL::TransposeOp>( new_ewise_op->getLoc(), op->getResult(0).getType(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
new_while_operands.push_back(it->getSecond().size); } auto new_while = builder.create<TF::WhileOp>( while_op.getLoc(), body.getFunctionType().getInputs(), new_while_operands, while_op->getAttrs()); for (const auto& entry : output_buffer_to_size) { (*buffer_to_size)[new_while.getResult(std::get<0>(entry))] = { new_while.getResult(std::get<1>(entry)), std::get<2>(entry)}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.cc
for (const auto& attr_def : op_reg_data->op_def.attr()) { registered_attrs.insert(attr_def.name()); } // Attributes are not in the registered attributes set will be ignored. for (auto& attr : inst->getAttrs()) { if (registered_attrs.find(attr.getName()) == registered_attrs.end()) { attrs_to_ignore->insert( absl::string_view(attr.getName().data(), attr.getName().size())); } }
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/lite/transforms/reduce_type_precision.cc
op.getLoc(), /*result=*/ mlir::cast<TensorType>(op.getResult().getType()) .clone(builder.getI4Type()), /*operand=*/op.getOperands(), op->getAttrs()); rewriter.replaceAllUsesWith(op.getResult(), new_gather_op.getResult()); return success(); } }; class ReduceTypePrecisionPass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0)