Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 25 for input_size (0.22 sec)

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

          scalar_mul(dilation_rate_value, scalar_sub(filter_size, one)), one);
    
      // output_size = (input_size + stride - 1) / stride
      Value output_size = scalar_div(
          scalar_add(input_size, scalar_sub(stride_value, one)), stride_value);
      // padding_needed = std::max(
      //     0,
      //     (output_size - 1) * stride + effective_filter_size - input_size)
      Value padding_needed = scalar_sub(
          scalar_add(effective_filter_size_op,
    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

            int64_t output_size;
            int64_t pad_low_int64;
            int64_t pad_high_int64;
            int64_t input_size = input_ty.getDimSize(dim);
            if (input_size == ShapedType::kDynamic) return failure();
            absl::Status status = tensorflow::GetWindowedOutputSizeVerbose(
                input_size, filter_ty.getDimSize(i), dilation, stride, padding,
                &output_size, &pad_low_int64, &pad_high_int64);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        Location loc = op.getLoc();
        Value scalar_zero = CreateI32SplatConst(loc, &rewriter, {}, 0);
    
        // Compute the input tensorlist's length and store it in `input_size`.
        IntegerType shape_dtype = rewriter.getIntegerType(32);
        auto input_size = rewriter.create<TF::TensorListLengthOp>(
            loc, tensorflow::GetTypeFromTFTensorShape({}, shape_dtype),
            op.getOperand(0));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

          for (int64_t i = 0; i < input_ty.getRank(); ++i) {
            int64_t slice_size = slice_sizes.getValues<APInt>()[i].getSExtValue();
            int64_t input_size = input_shape[i];
            if (slice_size != -1 && input_size != ShapedType::kDynamic &&
                slice_size > input_size) {
              return op.emitOpError() << "requires size[i] <= Di, even if begin[i] "
                                         "is unknown at compile time";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass.cc

                            std::vector<const Edge*> control_deps, Output* size) {
      // If slice_size[i] >= 0 then slice_size[i] = slice_size[i].
      //
      // If slice_size[i] == -1 then slice_size[i] = input_size[i] -
      // begin[i].
      //
      // If slice_size[i] < -1 then executing the slice will throw an error, and we
      // don't do anything here.  We've already filtered these cases out in
      // IsRewritableSlice.
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 13.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc

                               b.device());
        }
        return false;
      }
      if (a.input_size() != b.input_size()) {
        if (diff) {
          *diff = absl::StrCat(diff_preamble, " mismatch for node ", a.name(),
                               ", expected ", a.input_size(), " inputs got ",
                               b.input_size(), " expected:\n", a.DebugString(),
                               "\ngot:\n", b.DebugString());
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 113.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

                  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])) {
            align_corners = false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

    }
    
    static LogicalResult ComputeConvWindowedOutputSize(
        int64_t input_size, int64_t filter_size, int64_t dilation_rate,
        int64_t stride, tensorflow::Padding padding, int64_t* output_size) {
      int64_t pad_low;
      int64_t pad_high;
    
      tensorflow::Status status = tensorflow::GetWindowedOutputSizeVerbose(
          input_size, filter_size, dilation_rate, stride, padding, output_size,
          &pad_low, &pad_high);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

            mlir::dyn_cast<RankedTensorType>(input.getType());
        if (!input_type) {
          return failure();
        }
        const ArrayRef<int64_t> input_shape = input_type.getShape();
    
        const int input_size = input_shape.size();
    
        RankedTensorType begin_type = mlir::cast<RankedTensorType>(begin.getType());
        const ArrayRef<int64_t> begin_shape = begin_type.getShape();
        const int begin_dim = begin_shape.size();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td

        TF_Float32Tensor:$sorted_gains,
        TF_Float32Tensor:$embedding_table,
        TF_Int32Tensor:$num_minibatches_per_physical_sparse_core,
    
        ConfinedAttr<I64Attr, [IntMinValue<0>]>:$input_size,
        OptionalAttr<F32Attr>:$quantization_config_low,
        OptionalAttr<F32Attr>:$quantization_config_high,
        OptionalAttr<I64Attr>:$quantization_config_num_buckets,
        StrAttr:$table_name
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 90.5K bytes
    - Viewed (0)
Back to top