- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 84 for foo_attr (0.13 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
"whose value is non-negative">; def TFL_BoolTrue : AttrConstraint< CPred<"$_self.isa<BoolAttr>() && $_self.cast<BoolAttr>().getValue()">, "whose value is true">; def TFL_BoolFalse : AttrConstraint< CPred<"$_self.isa<BoolAttr>() && !$_self.cast<BoolAttr>().getValue()">, "whose value is false">; class TFL_StringEqualsTo<string value> : AttrConstraint<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
PatternRewriter& rewriter) const override { const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(call_op.getFAttr()); // removeAttr will return nullptr if no attribute was removed. if (!call_op->removeAttr(kQuantTraitAttrName) || !f_attr) { return failure(); } if (!f_attr.getValue().starts_with(kCompositeFuncPrefix)) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
// Note: this information is duplicated and can be removed from the proto // and here once MLIR bridge phase 2 doesn't fallback to the old bridge. auto attr = op.getFuncOp().getArgAttrOfType<mlir::BoolAttr>( index, replication_attr_name); arg->set_is_same_data_across_replicas(attr != nullptr && attr.getValue()); // Currently only support first dimension to be bounded dynamic.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc
TF_RETURN_IF_ERROR(ConvertLocation(locations[i], node_name, debug_info)); } } return absl::OkStatus(); } Status ConvertAttribute(const mlir::BoolAttr& attr, AttrValue* value) { value->set_b(attr.getValue()); return absl::OkStatus(); } Status ConvertAttribute(const mlir::IntegerAttr& attr, AttrValue* value) { value->set_i(attr.getInt());
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/compiler/mlir/lite/transforms/legalize_variables.cc
// 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); (void)applyPatternsAndFoldGreedily(module, std::move(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/quantization/tensorflow/passes/propagate_quantize_type.cc
} LogicalResult matchAndRewrite(TF::PartitionedCallOp op, PatternRewriter& rewriter) const override { const auto f_attr = mlir::dyn_cast<FlatSymbolRefAttr>(op.getFAttr()); StringRef function_name = f_attr.getValue(); if (!function_name.starts_with(kDequantizeFunctionName)) return failure(); llvm::SmallVector<Operation*> users(op->getUsers().begin(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.td
} def TF_DumpTensorOp : TF_Op<"DumpTensor", []> { let summary = "Dump tensor proto."; let arguments = (ins Arg<TF_Tensor>:$input, StrAttr:$log_dir_path, StrAttr:$file_name, BoolAttr:$enabled, StrAttr:$func_name, StrAttr:$node_name ); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<0>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
return entry_function_attr.getValue(); } else { TF::PartitionedCallOp call_op = dyn_cast_or_null<TF::PartitionedCallOp>(op); const auto f_attr = call_op.getFAttr().dyn_cast<FlatSymbolRefAttr>(); if (!f_attr) return std::nullopt; return f_attr.getValue(); } } class InsertCustomAggregationOpsPass : public PassWrapper<InsertCustomAggregationOpsPass,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
// destination type is not inferred and must be given explicitly. // // Preconditions: The given value must have a ShapedType. static Value CreateTFCastOpI32(OpBuilder *builder, Location loc, Value x, BoolAttr truncate) { auto x_type = mlir::dyn_cast_or_null<ShapedType>(x.getType()); if (!x_type) llvm_unreachable("unsupported type"); Type type = x_type.clone(builder->getI32Type());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/replace_cast_hacks_with_tf_xla_ops.cc
Value input, Value weight, Value input_zp, Value weight_zp, Value output, BoolAttr transpose_a, BoolAttr transpose_b) { // Transpose and constant-fold the weight if needed. if (transpose_b.getValue()) { Value perm = Create1DConstValue<int32_t>(builder, loc, {1, 0});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 47.1K bytes - Viewed (0)