Searched refs:FuseFullyConnectedAndReluX (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 405 struct FuseFullyConnectedAndReluX : public OpRewritePattern<ReluXOp> { struct
964 FuseFullyConnectedAndReluX<TFL::ReluOp, kRelu>, in runOnFunction()
965 FuseFullyConnectedAndReluX<TFL::Relu6Op, kRelu6>, in runOnFunction()
966 FuseFullyConnectedAndReluX<TFL::Relu1Op, kRelu1>, in runOnFunction()
976 FuseFullyConnectedAndAdd, FuseFullyConnectedAndReluX<TFL::ReluOp, kRelu>, in runOnFunction()
977 FuseFullyConnectedAndReluX<TFL::Relu6Op, kRelu6>, in runOnFunction()
978 FuseFullyConnectedAndReluX<TFL::Relu1Op, kRelu1>, in runOnFunction()
|