Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for perm_attr (0.08 sec)

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

        auto perm_type = tensorflow::GetTypeFromTFTensorShape(
            {static_cast<int>(perm.size())}, rewriter.getIntegerType(32));
        auto perm_attr =
            DenseElementsAttr::get(perm_type, llvm::ArrayRef<int>(perm));
        auto perm_op = rewriter.create<TF::ConstOp>(loc, perm_type, perm_attr);
    
        // Create tensor type for the transpose result.
        auto filter_type = mlir::cast<RankedTensorType>(filter.getType());
    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/transforms/lower_static_tensor_list.cc

        // as specified by element_dtype.
        RankedTensorType zero_type =
            tensorflow::GetTypeFromTFTensorShape({}, element_dtype);
        auto zero_attr = rewriter.getZeroAttr(zero_type);
        auto zero = rewriter.create<arith::ConstantOp>(loc, zero_type, zero_attr);
    
        rewriter.replaceOpWithNewOp<TF::FillOp>(op, result_type, list_shape, zero);
        return success();
      }
    };
    
    struct ConvertTensorListReserve
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
Back to top