Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for m_Constant (0.25 sec)

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

        if (!begin_attr_type ||
            !matchPattern(begin_attr, m_Constant(&begin_elem_attr))) {
          return failure();
        }
        if (!end_attr_type || !matchPattern(end_attr, m_Constant(&end_elem_attr))) {
          return failure();
        }
        if (!strides_attr_type ||
            !matchPattern(strides_attr, m_Constant(&strides_elem_attr))) {
          return failure();
        }
    
    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/tensorflow/transforms/lower_tf.cc

        ElementsAttr paddings;
        llvm::SmallVector<int64_t, 4> block_shape_ints;
        auto padded_shape = llvm::to_vector<4>(input_shape);
        if (matchPattern(op.getBlockShape(), m_Constant(&block_shape)) &&
            matchPattern(op.getPaddings(), m_Constant(&paddings))) {
          for (uint64_t i = 0; i < block_rank; i++) {
            int64_t paddings_sum =
                paddings.getValues<APInt>()[{i, 0}].getSExtValue() +
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

                         : op.emitOpError(error_info);
            }
          }
        }
    
        DenseIntElementsAttr dense_elem_attr;
        if (matchPattern(element_shape, m_Constant(&dense_elem_attr))) {
          // Note: It's technically unsafe to rewrite
          //     TensorListReserve(num_element, element_shape)
          // to
          //     Fill(Concat(num_element, element_shape), 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/quantization/tensorflow/passes/quantize_composite_functions.cc

     private:
      LogicalResult matchAndRewrite(quantfork::QuantizeCastOp q_op,
                                    PatternRewriter& rewriter) const override {
        DenseFPElementsAttr attr;
        if (!matchPattern(q_op.getArg(), m_Constant(&attr))) {
          return failure();
        }
    
        ShapedType tensor_qtype =
            mlir::cast<ShapedType>(q_op.getResult().getType());
        Attribute tensor_proto_attr = Quantize(attr, tensor_qtype);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

      }
      if (op_name == "tfl.reshape") {
        // Flattens reshape ops when more than one dimension shape operand is given.
        mlir::DenseIntElementsAttr shape_attr;
        if (matchPattern(op_state.operands[1], m_Constant(&shape_attr))) {
          auto shape_ty =
              op_state.operands[1].getType().dyn_cast<RankedTensorType>();
          if (shape_ty != nullptr && shape_ty.hasRank() && shape_ty.getRank() > 1) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
Back to top