HomeSort by relevance Sort by last modified time
    Searched refs:OpHint (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/python/
convert_test.py 235 custom = op_hint.OpHint("cool_activation")
263 custom = op_hint.OpHint("scale_and_bias_and_identity")
288 custom = op_hint.OpHint("add_test")
309 return x.op.node_def.attr[op_hint.OpHint.FUNCTION_INPUT_INDEX_ATTR].i
312 return x.op.node_def.attr[op_hint.OpHint.FUNCTION_OUTPUT_INDEX_ATTR].i
315 return x.op.node_def.attr[op_hint.OpHint.FUNCTION_SORT_INDEX_ATTR].i
323 custom = op_hint.OpHint("test_tag")
325 aggregate=op_hint.OpHint.AGGREGATE_STACK)
328 aggregate=op_hint.OpHint.AGGREGATE_STACK)
330 aggregate=op_hint.OpHint.AGGREGATE_STACK
    [all...]
op_hint.py 27 custom = tf.lite.OpHint("cool_activation")
45 OpHint is a helper that you use when defining a vanilla python function.
91 @_tf_export("lite.OpHint")
92 class OpHint(object):
111 # will be used on every Identity op that is involved in a total OpHint.
113 # Name of the OpHint function (cosmetic).
115 # UUID of the function (each OpHint gets a new uuid).
123 # has a lstm cell for each time step. Each one has a separate opHint, but a
127 # into a fused operand. Valid options are OpHint.AGGREGATE_FIRST,
128 # OpHint.AGGREGATE_LAST, OpHint.AGGREGATE_STACK
    [all...]
lite.py 44 from tensorflow.lite.python.op_hint import OpHint # pylint: disable=unused-import
    [all...]
  /external/tensorflow/tensorflow/lite/experimental/examples/lstm/
rnn.py 168 raise RuntimeError("OpHint dynamic rnn only supports control flow v2.")
193 tflite_wrapper = op_hint.OpHint(
rnn_cell.py 79 self._tflite_wrapper = op_hint.OpHint("UnidirectionalSequenceRnn")
257 self._tflite_wrapper = op_hint.OpHint("UnidirectionalSequenceLstm")

Completed in 276 milliseconds