- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for perm_attr (0.1 sec)
-
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) -
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)