TensorFlow函數(shù)教程：tf.lite.OpHint

tf.lite.OpHint函數(shù)

類 OpHint

別名：

• 類 tf.contrib.lite.OpHint
• 類 tf.lite.OpHint

定義在：tensorflow/lite/python/op_hint.py。

一個幫助構(gòu)建tflite函數(shù)調(diào)用的類。

它允許你采取一堆TensorFlow操作并注釋構(gòu)造，以便toco知道如何將其轉(zhuǎn)換為tflite。這在TensorFlow圖中嵌入了一個偽函數(shù)。這允許在較低級別的TensorFlow實現(xiàn)中嵌入高級API使用信息，以便以后可以替換替代實現(xiàn)。

本質(zhì)上，此偽操作中的任何“input”都被輸入到標識中，并且屬性在被構(gòu)成偽操作的組成操作使用之前被添加到該輸入。對從當前操作中導出的任何輸出執(zhí)行類似的過程。

__init__(
    function_name,
    **kwargs
)

創(chuàng)建一個OpHint。

參數(shù)：

• function_name：函數(shù)的名稱（tflite中的自定義操作名稱）
• **kwargs：函數(shù)的任何常量屬性的關(guān)鍵字參數(shù)。

子類

class OpHintArgumentTracker

方法

add_input(
    *args,
    **kwargs
)

將包裝的輸入?yún)?shù)添加到提示中。

參數(shù)：

• *args：輸入張量。
• **kwargs：
  • "tag"：用于對要聚合的多個參數(shù)進行分組。就像'cool_input'這樣的字符串?；旧峡梢詫⒍鄠€輸入添加到最終將被組合的并行操作的相同提示中。一個例子是static_rnn，它創(chuàng)建狀態(tài)或輸入的多個副本。
  • “aggregate”：聚合策略，僅對標簽不是none的有效?？山邮艿闹禐镺pHint.AGGREGATE_FIRST，OpHint.AGGREGATE_LAST和OpHint.AGGREGATE_STACK。
  • “index_override”：要使用的全局索引。這對應于將生成的最終存根中的參數(shù)順序。

返回：

包裝的輸入張量。

add_inputs(
    *args,
    **kwargs
)

向函數(shù)調(diào)用添加一系列輸入。

參數(shù)：

• *args：要轉(zhuǎn)換的輸入列表（應為Tf.Tensor）。
• **kwargs：這允許“names”應該是名稱列表。

返回：

包裝的輸入（具有附加元數(shù)據(jù)的身份標記）。這些也是tf.Tensor的。

add_output(
    *args,
    **kwargs
)

將一個包裝的輸出參數(shù)添加到提示中。

參數(shù)：

• *args：輸出張量。
• **kwargs：
  • “tag”：一個標記，用于對要聚合的多個參數(shù)進行分組。就像'cool_input'這樣的字符串?；旧峡梢詫⒍鄠€輸入添加到最終將被組合的并行操作的相同提示中。一個例子是static_rnn，它創(chuàng)建狀態(tài)或輸入的多個副本。
  • “aggregate”：聚合策略，僅對標簽不是none的有效。可接受的值為OpHint.AGGREGATE_FIRST，OpHint.AGGREGATE_LAST和OpHint.AGGREGATE_STACK。
  • “index_override”：要使用的全局索引。這對應于將生成的最終存根中的參數(shù)順序。

返回：

包裝的輸出張量。

add_outputs(
    *args,
    **kwargs
)

將一系列輸出添加到函數(shù)調(diào)用中。

參數(shù)：

• *args：要轉(zhuǎn)換的輸出列表（應為tf.Tensor）。

返回：

包裝的輸出（具有附加元數(shù)據(jù)的標識站）。這些也是tf.Tensor的。

類成員