Home
last modified time | relevance | path

Searched defs:reshape_type (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Doptimize.cc61 auto reshape_type = reshape_op.output().getType().cast<ShapedType>(); in matchAndRewrite() local
Dtpu_space_to_depth_pass.cc198 auto reshape_type = RankedTensorType::get( in GetReshapeOpForConv2DFilter() local
Dlegalize_hlo.cc1321 auto reshape_type = reshape_op.getType().cast<RankedTensorType>(); in MatchReshapedIota() local
2271 auto reshape_type = RankedTensorType::get(reshape_shape, element_type); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tfl.cc1504 RankedTensorType reshape_type = in matchAndRewrite() local