Searched refs:OpHint (Results 1 – 13 of 13) sorted by relevance
92 class OpHint(object): class278 OpHint.FUNCTION_NAME_ATTR,282 OpHint.FUNCTION_UUID_ATTR,287 identity_op.op._set_attr(OpHint.FUNCTION_LEVEL_ATTR,291 OpHint.CHILDREN_INPUTS_MAPPINGS,298 OpHint.FUNCTION_SORT_INDEX_ATTR,302 OpHint.FUNCTION_AGGREGATE_ATTR,339 self._inputs = OpHint.OpHintArgumentTracker(341 OpHint.FUNCTION_INPUT_INDEX_ATTR, level, self._children_inputs_mappings)342 self._outputs = OpHint.OpHintArgumentTracker([all …]
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].i312 return x.op.node_def.attr[op_hint.OpHint.FUNCTION_OUTPUT_INDEX_ATTR].i315 return x.op.node_def.attr[op_hint.OpHint.FUNCTION_SORT_INDEX_ATTR].i323 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 …]
44 from tensorflow.lite.python.op_hint import OpHint # pylint: disable=unused-import
1 path: "tensorflow.lite.OpHint"3 is_instance: "<class \'tensorflow.lite.python.op_hint.OpHint\'>"
1 path: "tensorflow.lite.OpHint.OpHintArgumentTracker"
8 name: "OpHint"
106 [OpHint](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/python/op_hint.py)112 ### What's OpHint114 `OpHint` is essentially `Identity` op that is inserted after input tensors and120 add op followed by an activation op (graph on the letf), we use `OpHint` to
79 self._tflite_wrapper = op_hint.OpHint("UnidirectionalSequenceRnn")257 self._tflite_wrapper = op_hint.OpHint("UnidirectionalSequenceLstm")
193 tflite_wrapper = op_hint.OpHint(
189 In TensorFlow you can use the `tf.lite.OpHint` class to encapsulate groups of194 [OpHint code](https://github.com/tensorflow/tensorflow/blob/master/tensorflow/lite/python/op_hint.p…