- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for BoolAttr (0.49 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
>; // This decomposition is only correct inside XLA as it ignores use_locking // attribute. def DecomposeResourceApplyGradientDescentOp : Pat< (TF_ResourceApplyGradientDescentOp:$src_op $resource, $alpha, $delta, BoolAttr:$_), (TF_AssignVariableOp $resource, (TF_SubOp (CreateTFReadVariableOp $src_op, $alpha, $resource), (TF_MulOp $alpha, $delta) ), (CreateConstBoolAttrFalse) )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td
def LegalizeMin : Pat< (TF_MinOp $arg0, $axes, BoolAttr:$arg2), (TFL_ReduceMinOp $arg0, (CreateTFCastToInt32Op $axes), $arg2)>; def LegalizeMax : Pat< (TF_MaxOp $arg0, $axes, BoolAttr:$arg2), (TFL_ReduceMaxOp $arg0, (CreateTFCastToInt32Op $axes), $arg2)>; def LegalizeProd : Pat< (TF_ProdOp $arg0, $axes, BoolAttr:$arg2), (TFL_ReduceProdOp $arg0, (CreateTFCastToInt32Op $axes), $arg2)>;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 04 13:30:42 UTC 2024 - 28.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/QuantOps.td
I64Attr:$num_bits, // Quantization range starts from 0 or 1; starts from 1 if true. DefaultValuedOptionalAttr<BoolAttr, "false">:$narrow_range, // The sign of the quantization. DefaultValuedOptionalAttr<BoolAttr, "false">:$is_signed ); let results = (outs F32Tensor:$outputs ); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 13 12:46:08 UTC 2022 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/QuantOps.td
I64Attr:$num_bits, // Quantization range starts from 0 or 1; starts from 1 if true. DefaultValuedOptionalAttr<BoolAttr, "false">:$narrow_range, // The sign of the quantization. DefaultValuedOptionalAttr<BoolAttr, "false">:$is_signed ); let results = (outs F32Tensor:$outputs ); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 09 03:10:59 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
if (callable->hasAttr("tf._XlaMustCompile")) return true; auto noinline_attr_name = absl::StrCat("tf.", tensorflow::kNoInlineAttr); if (auto noinline_attr = callable->getAttrOfType<BoolAttr>(noinline_attr_name)) return !noinline_attr.getValue(); return true; } // Returns if its legal to inline 'src' region into the 'dest' region // attached to a TF operation.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K 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/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) -
tensorflow/compiler/mlir/lite/utils/lstm_utils.cc
mlir::TFL::LSTMKernelTypeAttr::get(builder_.getContext(), mlir::TFL::LSTMKernelType::FULL), /*asymmetric_quantize_inputs=*/mlir::BoolAttr(), /*input_to_input_intermediate=*/mlir::TypeAttr(), /*input_to_forget_intermediate=*/mlir::TypeAttr(), /*input_to_cell_intermediate=*/mlir::TypeAttr(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 36.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
} } else if (llvm::isa<TF::BatchMatMulV2Op>(op)) { if (const auto adj_y = op->getAttrOfType<BoolAttr>("adj_y")) { is_supported_affine_op = !adj_y.getValue(); } } else if (llvm::isa<TF::MatMulOp>(op)) { if (const auto adj_y = op->getAttrOfType<BoolAttr>("transpose_b")) { is_supported_affine_op = !adj_y.getValue(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0)