- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for IntegerAttr (0.27 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
window_dimensions.getValues<IntegerAttr>()[i].getInt()); } for (auto i = 0; i < window_strides.size(); ++i) { window_strides_vec.push_back( window_strides.getValues<IntegerAttr>()[i].getInt()); } for (auto i = 0; i < base_dilations.size(); ++i) { base_dilations_vec.push_back( base_dilations.getValues<IntegerAttr>()[i].getInt()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
if (elem_type.isSignlessInteger() || elem_type.isUnsignedInteger()) { auto start_attr = start_tensor.getValues<IntegerAttr>()[0]; auto limit_attr = limit_tensor.getValues<IntegerAttr>()[0]; auto delta_attr = delta_tensor.getValues<IntegerAttr>()[0]; int num_elements; if (elem_type.isUnsignedInteger()) { uint64_t start = start_attr.getUInt();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.cc
Type elem_type = getType().cast<ShapedType>().getElementType(); if (elem_type.isSignlessInteger()) { auto start_attr = start_tensor.getValues<IntegerAttr>()[0]; auto limit_attr = limit_tensor.getValues<IntegerAttr>()[0]; auto delta_attr = delta_tensor.getValues<IntegerAttr>()[0]; const int num_elements = GetLengthOfRange( start_attr.getInt(), limit_attr.getInt(), delta_attr.getInt());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 169.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
DenseIntElementsAttr axis_attr; if (!matchPattern(op.getAxis(), m_Constant(&axis_attr))) return failure(); if (axis_attr.getNumElements() != 1) return failure(); int64_t axis = axis_attr.getSplatValue<IntegerAttr>().getValue().getSExtValue(); // TODO(ezhulenev): Compute axis from rank. e.g. It might be common to concat // on the channels dim for NCHW layout as axis=-2. if (axis < 0) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
stride_w = CastI64ToI32((*window_strides)[2]).value(); } auto stride_h_attr = IntegerAttr::get(rewriter.getI32Type(), stride_h); auto stride_w_attr = IntegerAttr::get(rewriter.getI32Type(), stride_w); ArrayRef<int64_t> window_dims = op.getWindowDimensions(); auto window_w_attr = IntegerAttr::get(rewriter.getI32Type(), CastI64ToI32(window_dims[2]).value());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
float_spl && sgn_cst_spl) { return IsSign(float_spl.getValue(), sgn_cst_spl.getValue()); } if (auto int_spl = llvm::dyn_cast_if_present<IntegerAttr>(splat), sgn_cst_spl = llvm::dyn_cast_if_present<IntegerAttr>(sgn_splat); int_spl && sgn_cst_spl) { return IsSign(int_spl.getValue(), sgn_cst_spl.getValue()); } if (mlir::isa<DenseFPElementsAttr>(float_or_int)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
"whose size is at most " # n>; // Ensures the given integer attribute has the given value. class TFL_IntEqualsTo<int n> : AttrConstraint< CPred<"$_self.isa<IntegerAttr>() && " "$_self.cast<IntegerAttr>().getInt() == " # n>, "whose value is " # n>; // Ensures the given LSTMKernelType attribute has the given value. class TFL_LSTMKernelTypeEqualsTo<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/tensorflow/translate/import_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)