Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for input_size (0.17 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let description = [{
        A recurrent neural network specified by an RNN cell. This Op takes in input
        in a format {batch_size, seq_len, input_size} or
        {seq_len, batch_size, input_size} if it's time-majored.
    
        It implements the following operation for
        each element in the sequence s = 1...sequence_length:
          outputs[s] = state = activation(RNNOp(inputs[s]))
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top