Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for begin_shape (0.47 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        const ArrayRef<int64_t> begin_shape = begin_type.getShape();
        const int begin_dim = begin_shape.size();
    
        if (begin_dim != 1) return failure();
    
        // The ellipsis fill might exceed the current output shape because we are
        // also taking account of any to-be-inserted new axes.
        const int ellipsis_filled_dim_size =
            input_size - begin_shape[0] + 1 + absl::popcount(new_axis_mask);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

    ArrayRef<int64_t> GetRankedTensorShape(Value value) {
      return mlir::cast<RankedTensorType>(value.getType()).getShape();
    }
    
    Value SliceRankedTensor(OpBuilder* builder, Value input,
                            ArrayRef<int64_t> begin_shape,
                            ArrayRef<int64_t> begin_values,
                            ArrayRef<int64_t> size_shape,
                            ArrayRef<int64_t> size_values,
                            mlir::Location location) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

              rewriter.create<TFL::MinimumOp>(op->getLoc(), cast, upper_limit_cst);
        }
    
        Value concatenated = start_indices[0];
        if (rank > 1) {
          SmallVector<int64_t> begin_shape{rank};
          Type begin_type = operand_type.cloneWith(begin_shape, i64_type);
          concatenated = rewriter.create<TFL::ConcatenationOp>(
              op->getLoc(), begin_type, start_indices, /*axis=*/0,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
Back to top