Home
last modified time | relevance | path

Searched defs:reshape_buffer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_reorder_axes.cc52 auto& reshape_buffer = in CreateReshapeFromReorderAxes() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc1170 Value reshape_buffer = rewriter.create<memref::ExpandShapeOp>( in matchAndRewrite() local
1187 Value reshape_buffer = isCollapsing in matchAndRewrite() local