Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for spatial_dims (0.18 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

                                         Value &padding, int num_dims) {
      ShapedType input_shape = mlir::cast<ShapedType>(input.getType());
      SmallVector<int64_t> spatial_dims(num_dims - 2);
      absl::c_iota(spatial_dims, 1);
      bool has_dynamic_spatial_dim = absl::c_any_of(
          spatial_dims,
          [&input_shape](int64_t dim) { return input_shape.isDynamicDim(dim); });
      if (conv_padding.strref() == "SAME" && has_dynamic_spatial_dim) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

          // Prepare metadata indexed by spatial_dim for computing pad_before
          // and pad_after.
          int64_t input_size = input_shape[spatial_dim];
          if (input_size == ShapedType::kDynamic) return failure();
          int64_t output_size = out_backprop_ty.getDimSize(spatial_dim);
          if (output_size == ShapedType::kDynamic) return failure();
          int64_t filter_size = filter_ty.getDimSize(i);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top