- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for padding_values (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
int padding_after = padding_before + offset; padding_values[2 * i] = padding_before; padding_values[2 * i + 1] = padding_after; } } if (input_zp_value == 0 || absl::c_all_of(padding_values, [](int v) { return v == 0; })) { padding = CreateConstValue<int32_t>( builder, loc, {num_dims - 2, 2}, SmallVector<int32_t>(padding_values.begin() + 2,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/composite_avg_pool.cc
} ShapedType GetPadOpType(CompositeOp op) { const TorchAvgPoolData pool = GetTorchAvgPoolData(op); const auto padding_values = GetPadOpPaddingValues(pool); const int64_t h = pool.h_in + padding_values[0] + padding_values[1]; const int64_t w = pool.w_in + padding_values[2] + padding_values[3]; llvm::SmallVector<int64_t> shape = {pool.n, h, w, pool.c};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
if (!padding_attr) { return {0, 0, 0, 0}; } auto padding_values = padding_attr.getValues<int64_t>(); // Padding has [[h_low, h_high], [w_low, w_high]] format. // https://github.com/openxla/stablehlo/blob/main/docs/spec.md#convolution. return {padding_values[0], padding_values[1], padding_values[2], padding_values[3]}; } // Returns the input and output dimensions, respectively.
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/transforms/prepare_tf.cc
auto get_int = [](Attribute attr) { return mlir::cast<IntegerAttr>(attr).getInt(); }; SmallVector<int32_t> padding_values(padding_attr_array.size()); for (int i = 0; i < padding_attr_array.size(); i++) { padding_values[i] = static_cast<int32_t>(get_int(padding_attr_array[i])); } RankedTensorType padding_attr_type = tensorflow::GetTypeFromTFTensorShape(
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/lite/stablehlo/transforms/legalize_hlo.cc
/*align_corners=*/true, lhs_dilation_values[0], padding_values[0], window_strides_values[0], input_sizes[0], output_sizes[0])) { align_corners = true; return success(); } if (can_convert_to_bilinear( /*align_corners=*/false, lhs_dilation_values[0], padding_values[0], window_strides_values[0], input_sizes[0], output_sizes[0])) {
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/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
CompileTimeConstantOperand<TF::ListDiffOp, 0, 1>, // $x, $y // $k, $padding_value CompileTimeConstantOperand<TF::MatrixDiagPartV3Op, 1, 2>, // $k, $num_rows, $num_cols, $padding_value CompileTimeConstantOperand<TF::MatrixDiagV2Op, 1, 2, 3, 4>, // $k, $num_rows, $num_cols, $padding_value CompileTimeConstantOperand<TF::MatrixDiagV3Op, 1, 2, 3, 4>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
// Verify padding_value is a tensor with all 0s. mlir::Value padding_value = tf_matrix_diag_v2_or_v3_op.getPaddingValue(); mlir::Type element_type = mlir::cast<ShapedType>(padding_value.getType()).getElementType(); if (mlir::isa<FloatType>(element_type)) { DenseFPElementsAttr padding_attr; if (!matchPattern(padding_value, m_Constant(&padding_attr)) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
auto padding_type = RankedTensorType::get({4, 2}, builder->getIntegerType(32)); auto paddings = DenseIntElementsAttr::get(padding_type, values); auto paddings_value = builder->create<TF::ConstOp>(filter.getLoc(), paddings); std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, filter_shape[1] + pad_w, filter_shape[2],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
} SetOpAttrValueScalar(ctx, op, i_list_values, "ksize", status); SetOpAttrValueScalar(ctx, op, i_list_values, "strides", status); tensorflow::AttrValue padding_value; *padding_value.mutable_s() = "VALID"; tensorflow::SetOpAttrValueScalar(ctx, op, padding_value, "padding", status); tensorflow::AttrValue data_format_value; *data_format_value.mutable_s() = "NHWC";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
if (!paddings_op) return failure(); auto paddings_value = paddings_op.getValue().dyn_cast<DenseElementsAttr>(); if (!paddings_value || paddings_value.getNumElements() != permutation.size() * 2) return failure(); SmallVector<int32_t, 8> shuffled_paddings(paddings_value.getNumElements()); for (const auto &index_pair : llvm::enumerate(paddings_value.getValues<APInt>())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)