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

tf.lite.OpHint函數(shù)

類 OpHint

別名：

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

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

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

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

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

__init__(
    function_name,
    **kwargs
)

創(chuàng)建一個(gè)OpHint。

參數(shù)：

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

子類

class OpHintArgumentTracker

方法

add_input(
    *args,
    **kwargs
)

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

參數(shù)：

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

返回：

包裝的輸入張量。

add_inputs(
    *args,
    **kwargs
)

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

參數(shù)：

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

返回：

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

add_output(
    *args,
    **kwargs
)

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

參數(shù)：

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

返回：

包裝的輸出張量。

add_outputs(
    *args,
    **kwargs
)

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

參數(shù)：

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

返回：

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

類成員