একটি main() ফাংশন সহ একটি মডিউল হিসাবে প্রকাশ করা একটি নির্বিচারে MLIR গণনা মোড়ানো হয়।
এই অপারেশনটির কোনো সংশ্লিষ্ট কার্নেল নেই এবং এটি একটি নিয়মিত TensorFlow সেশনে কার্যকর করার উদ্দেশ্যে নয়। পরিবর্তে এটি পরীক্ষার জন্য বা বিশেষ ক্ষেত্রে ব্যবহার করার উদ্দেশ্যে যেখানে একজন ব্যবহারকারী কাস্টম টুলিং প্রক্রিয়াকরণের অভিপ্রায়ে একটি TensorFlow গ্রাফের মাধ্যমে কাস্টম MLIR গণনা পাস করতে চান (যখন একটি ভিন্ন পরিবেশকে লক্ষ্য করে, যেমন TensorFlow lite)। এমএলআইআর মডিউলে একটি প্রধান() ফাংশন থাকতে পারে যা একটি এন্ট্রি পয়েন্ট হিসাবে ব্যবহার করা হবে। ক্রিয়াকলাপের ইনপুট প্রধান() ফাংশনে যুক্তি হিসাবে প্রেরণ করা হবে এবং আউটপুটগুলিতে ম্যাপ করা প্রধান ফাংশনের প্রত্যাবর্তিত মানগুলি। উদাহরণ ব্যবহার:
{@code
import tensorflow as tf
from tensorflow.compiler.mlir.tensorflow.gen_mlir_passthrough_op import mlir_passthrough_op
mlir_module = '''python
func @main(%arg0 : tensor<10xf32>, %arg1 : tensor<10xf32>) -> tensor<10x10xf32> {
%add = "magic.op"(%arg0, %arg1) : (tensor<10xf32>, tensor<10xf32>) -> tensor<10x10xf32>
return %ret : tensor<10x10xf32>
}
'''পাবলিক পদ্ধতি
| স্ট্যাটিক MlirPassthroughOp | |
| ইটারেটর< অপারেন্ড <অবজেক্ট>> | |
| তালিকা< আউটপুট <?>> | আউটপুট () |
উত্তরাধিকারসূত্রে প্রাপ্ত পদ্ধতি
পাবলিক পদ্ধতি
পাবলিক স্ট্যাটিক MlirPassthroughOp তৈরি করুন ( স্কোপ স্কোপ, Iterable< Operand <?>> inputs, String mlirModule, List<Class<?>> Toutputs)
একটি নতুন MlirPassthroughOp অপারেশন মোড়ানো একটি ক্লাস তৈরি করার কারখানা পদ্ধতি।
পরামিতি
| সুযোগ | বর্তমান সুযোগ |
|---|
রিটার্নস
- MlirPassthroughOp এর একটি নতুন উদাহরণ