49 #define FOR_ALL_INPUT_AND_WEIGHT_TENSORS(ACTION) \  argument
50     ACTION(Input)                                \
51     ACTION(InputToInputWeights)                  \
52     ACTION(InputToCellWeights)                   \
53     ACTION(InputToForgetWeights)                 \
54     ACTION(InputToOutputWeights)                 \
55     ACTION(RecurrentToInputWeights)              \
56     ACTION(RecurrentToCellWeights)               \
57     ACTION(RecurrentToForgetWeights)             \
58     ACTION(RecurrentToOutputWeights)             \
59     ACTION(CellToInputWeights)                   \
60     ACTION(CellToForgetWeights)                  \
61     ACTION(CellToOutputWeights)                  \
62     ACTION(InputGateBias)                        \
63     ACTION(CellGateBias)                         \
64     ACTION(ForgetGateBias)                       \
65     ACTION(OutputGateBias)                       \
66     ACTION(ProjectionWeights)                    \
67     ACTION(ProjectionBias)                       \
68     ACTION(OutputStateIn)                        \
69     ACTION(CellStateIn)
71 #define FOR_ALL_LAYER_NORM_WEIGHTS(ACTION) \  argument
72     ACTION(InputLayerNormWeights)          \
73     ACTION(ForgetLayerNormWeights)         \
74     ACTION(CellLayerNormWeights)           \
75     ACTION(OutputLayerNormWeights)
78 #define FOR_ALL_OUTPUT_TENSORS(ACTION) \  argument
79     ACTION(ScratchBuffer)              \
80     ACTION(OutputStateOut)             \
81     ACTION(CellStateOut)               \
82     ACTION(Output)