Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1379 llvm::SmallVector<int64_t, 8> output_ty_shape; in GetReshapeOutputType() local
1380 output_ty_shape.reserve(shape_attr.getNumElements()); in GetReshapeOutputType()
1404 output_ty_shape.push_back(size); in GetReshapeOutputType()
1408 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
1434 output_ty_shape[unknown_index] = missing_dim; in GetReshapeOutputType()
1437 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc607 llvm::SmallVector<int64_t, 8> output_ty_shape; in GetReshapeOutputType() local
608 output_ty_shape.reserve(shape_attr.getNumElements()); in GetReshapeOutputType()
629 output_ty_shape.push_back(size); in GetReshapeOutputType()
633 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()
659 output_ty_shape[unknown_index] = missing_dim; in GetReshapeOutputType()
662 output_ty = RankedTensorType::get(output_ty_shape, element_ty); in GetReshapeOutputType()