Searched defs:enable_stitch_fusion (Results 1 – 3 of 3) sorted by relevance
70 def get_softmax_output(x, y, enable_stitch_fusion): argument
66 def get_dropoutgrad_reducesum_output(x, y, z, enable_stitch_fusion): argument
63 def get_layernorm_output(x, y, z, enable_stitch_fusion): argument