最近在学XLA。XLA（Accelerated Linear Algebra）是一个特定于领域的线性代数编译器，可以加速TensorFlow模型，而无需更改源代码。由于文献资料的缺乏，有些构件很难系统地学习。
我试图理解匹配器，如[m::MaximumAnyOrder][1]，m::Broadcast。但https://github.com/tensorflow/tensorflow/blob/master/tensorflow/compiler/xla/service/gpu/cudnn_fused_conv_rewriter.cc#L607-L608我没有这些API的函数定义或文档。似乎它们不在XLA的代码库中。
有哪位Maven可以帮助我如何找到API定义或文档，以便更好地理解它们是如何工作的？
1条答案
按热度按时间4uqofj5v1#
MaximumAnyOrder是在www.example.com中声明的宏扩展https://github.com/tensorflow/tensorflow/blob/397f997d2882279f17377ffcf5e28855194310bc/tensorflow/compiler/xla/service/pattern_matcher.h#L2389。我不认为有太多关于此类函数的文档，希望您阅读代码。