- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 37 for get_attr (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
TF::ConstOp GetI64ConstantTensor(PatternRewriter &rewriter, ArrayRef<int64_t> values, Location location) { auto cst_attr = rewriter.getI64TensorAttr(values); return rewriter.create<TF::ConstOp>(location, cst_attr.getType(), cst_attr); } // Rewrites broadcast->reshape to a reshape->broadcast that reduces // the rank of the input and output of the broadcast.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
src/cmd/vendor/golang.org/x/sys/unix/zsymaddr_zos_s390x.s
MOVD $·Fremovexattr(SB), R8 MOVD R8, ret+0(FP) RET // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT TEXT ·get_FgetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Fgetxattr(SB), R8 MOVD R8, ret+0(FP) RET // THIS FILE IS GENERATED BY THE COMMAND AT THE TOP; DO NOT EDIT TEXT ·get_FsetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Fsetxattr(SB), R8 MOVD R8, ret+0(FP) RET
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 08 16:12:58 UTC 2024 - 8.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/folders.cc
return failure(); } res.push_back(signed_lhs_val / signed_rhs_val); } auto res_attr = DenseElementsAttr::get( const_oprs[0].getType().cast<RankedTensorType>(), res); rewriter.replaceOpWithNewOp<stablehlo::ConstantOp>(adaptor.value().Op(), res_attr); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 06:11:55 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
} } if (!in_order) { // Functions do not get reused in practice, so skip the check for if the // callee has been updated. StringAttr callee_sym = cluster_func_op.getFuncAttr().getAttr(); MoveResourceArgsToEnd(symtab.lookup<func::FuncOp>(callee_sym)); } builder.setInsertionPoint(cluster_func_op); auto xla_launch_op = builder.create<TF::XlaLaunchOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) { if (!mlir::cast<BoolAttr>(legalize_tfl_variables_attr).getValue()) return; } RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc
LogicalResult matchAndRewrite(mlir::stablehlo::CustomCallOp op, PatternRewriter &rewriter) const override { auto backendConfig = mlir::dyn_cast<DictionaryAttr>(op->getAttr("composite.backend_config")); if (!backendConfig) return op->emitError( "custom_call has no 'composite.backend_config' attribute or the " "attribute is not a dictionary");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
cluster_op->setAttr( builder->getStringAttr(kFuncAttr), mlir::SymbolRefAttr::get(builder->getContext(), outlined_func.getName())); builder->setInsertionPoint(cluster_op); auto cluster_func_op = builder->create<tf_device::ClusterFuncOp>( cluster_op.getLoc(), outlined_func.getFunctionType().getResults(), live_ins.getArrayRef(), cluster_op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K 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) -
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)