- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 26 for getFAttr (0.13 sec)
-
tensorflow/compiler/mlir/lite/quantization/lite/tfl_to_std.cc
auto dcast = b.create<DequantizeOp>(dq.getLoc(), dq.getResult().getType(), dq.getArg()); dq.getResult().replaceAllUsesWith(dcast); if (auto extra_attr = op->getAttr(mlir::quant::kVolatileOpAttrName)) { dcast->setAttr(mlir::quant::kVolatileOpAttrName, extra_attr); } dq.erase(); } else if (auto q = llvm::dyn_cast<quantfork::QuantizeCastOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 02:50:01 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
LogicalResult matchAndRewrite(TF::CustomAggregatorOp op, PatternRewriter &rewriter) const override { FloatAttr min = mlir::dyn_cast_or_null<FloatAttr>(op->getAttr("min")); FloatAttr max = mlir::dyn_cast_or_null<FloatAttr>(op->getAttr("max")); // When there are no min and max attributes, remove op. if (min == nullptr || max == nullptr) { op.getOutput().replaceAllUsesWith(op.getInput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op.cc
OP_REQUIRES_OK(context, context->GetAttr("output_file_path", &output_file_path)); OP_REQUIRES_OK(context, context->env()->NewWritableFile(output_file_path, &output_file_)); OP_REQUIRES_OK(context, context->GetAttr("ids", &ids_)); OP_REQUIRES_OK(context, context->GetAttr("calibration_methods",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 01:31:23 UTC 2024 - 8K bytes - Viewed (0) -
src/cmd/vendor/golang.org/x/sys/unix/symaddr_zos_s390x.s
MOVD $·Pipe2(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_FlockAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Flock(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_GetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Getxattr(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_NanosleepAddr(SB), NOSPLIT|NOFRAME, $0-8 MOVD $·Nanosleep(SB), R8 MOVD R8, ret+0(FP) RET TEXT ·get_SetxattrAddr(SB), NOSPLIT|NOFRAME, $0-8
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed May 08 16:12:58 UTC 2024 - 1.6K 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/jit/xla_device_ops.cc
<< type_string() << " on an XLA device. This should never happen."; } XlaAssignVariableOp::XlaAssignVariableOp(OpKernelConstruction* c) : OpKernel(c) { OP_REQUIRES_OK(c, c->GetAttr("dtype", &dtype_)); } void XlaAssignVariableOp::Compute(OpKernelContext* context) { OP_REQUIRES(context, dtype_ == context->input(1).dtype(), errors::InvalidArgument(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/infeed_ops_xla_adjust_layout.cc
OpBuilder builder(op.getContext()); SmallVector<Type> result_types(op.getResultTypes().begin(), op.getResultTypes().end()); if (!op->getAttr("layout")) { auto layout = mlir::GetTPUInfeedLayout(result_types, builder); if (failed(layout)) return; op->setAttr("layout", layout.value()); } } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K 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/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/launch_to_device_attribute.cc
if (op->getDialect() != tf_dialect) return WalkResult::advance(); if (parallel_group_attr) { op->setAttr(TF::kParallelExecAnnotation, parallel_group_attr); } auto device_attr = op->getAttr(kDeviceAttr); if (!device_attr) { op->setAttr(kDeviceAttr, launch.getDeviceAttr()); return WalkResult::advance(); } if (auto device_str_attr = mlir::dyn_cast<StringAttr>(device_attr)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0)