- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 72 for BoolAttr (0.16 sec)
-
tensorflow/compiler/jit/xla_kernel_creator_test.cc
std::unique_ptr<ProcessFunctionLibraryRuntime> pflr_; std::unique_ptr<OpKernel> kernel_; }; AttrValue BoolAttr(bool b) { AttrValue v; v.set_b(b); return v; } TEST_F(XlaKernelCreatorTest, OneFloatOneResourceArgument) { FunctionDef fdef = XTimesY(); (*fdef.mutable_attr())["_XlaMustCompile"] = BoolAttr(true); Init({fdef}); XlaKernelCreator xla_kernel_creator; auto callsite =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 16 01:39:55 UTC 2023 - 5.7K bytes - Viewed (0) -
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/stablehlo/transforms/composite_lowering_patterns.td
(TFL_HardSwishOp $input)>; def IsNchwLayoutOp: Constraint<CPred< "$0.get(\"is_nchw_op\") && $0.get(\"is_nchw_op\").dyn_cast<BoolAttr>() " "== mlir::BoolAttr::get($_builder.getContext(), true)">>; def IsNhwcLayoutOp: Constraint<Neg<IsNchwLayoutOp.predicate>>; // TODO(b/343278954): Move the creation of transposes to a separate prepare pass
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_utils.td
// Receives a composite DictionaryAttr as an argument and checks if has a // BoolAttr with the name `attr_name` and value `val`. class IsBoolCompositeAttribute <string attr_name, string val>: IsCompositeAttribute<attr_name, BoolAttr, val>; // Receives a composite DictionaryAttr as an argument and checks if has a // StrAttr with the name `attr_name` and value `val`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 2.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/tensorflow/ir/tf_ops_tensor_helper.h
Type DeduceEqualCmpOpType(Builder *builder, Location loc, Value x, Value y, BoolAttr incompatible_shape_error); Type InferReductionOpType(Value input, Value reduction_indices, BoolAttr keep_dims); // Verifies that the given types are cast compatible. If not, emits appropriate
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K 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/ir/tf_generated_ops.td
DefaultValuedOptionalAttr<BoolAttr, "false">:$adj_x, DefaultValuedOptionalAttr<BoolAttr, "false">:$adj_y, DefaultValuedOptionalAttr<BoolAttr, "false">:$grad_x, DefaultValuedOptionalAttr<BoolAttr, "false">:$grad_y ); let results = (outs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)