Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform_patterns.cc80 SmallVector<int32_t, 4> new_shape_array_i32; in InsertReshapeOp() local
82 new_shape_array_i32.push_back(static_cast<int32_t>(size)); in InsertReshapeOp()
85 mlir::DenseIntElementsAttr::get(reshape_shape_type, new_shape_array_i32); in InsertReshapeOp()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc1496 SmallVector<int32_t, 4> new_shape_array_i32; in matchAndRewrite() local
1498 new_shape_array_i32.push_back(static_cast<int32_t>(size)); in matchAndRewrite()
1503 RankedTensorType::get(new_shape_array_i32.size(), in matchAndRewrite()
1505 new_shape_array_i32)); in matchAndRewrite()