Home
last modified time | relevance | path

Searched refs:final_reshape_op (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc306 auto final_reshape_op = in HandleConv2DFilter() local
309 conv2d.setOperand(1, final_reshape_op); in HandleConv2DFilter()
396 auto final_reshape_op = in HandleConv2DBackPropFilter() local
401 final_reshape_op, &builder); in HandleConv2DBackPropFilter()