Searched refs:createReshapeOp (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 64 TF::ReshapeOp ConvertTFBatchMatMulOp<BatchMatMulOpType>::createReshapeOp( in createReshapeOp() function in mlir::TF::ConvertTFBatchMatMulOp
89 auto reshape_op = createReshapeOp(value, {batch_size, num_rows, num_cols}, in sliceInput()
114 auto squeeze_op = createReshapeOp(slice_op.output(), {num_rows, num_cols}, in sliceInput()
303 auto reshape_op = createReshapeOp(pack_op.output(), result_shape, in matchAndRewrite()
|
| D | einsum.cc | 71 TF::ReshapeOp createReshapeOp(Value value, ArrayRef<int64_t> shape, in createReshapeOp() function
292 *lhs = createReshapeOp(*lhs, lhs_shape, lhs_type.getElementType(), loc, in reshapeForBatchMatmul()
294 *rhs = createReshapeOp(*rhs, rhs_shape, rhs_type.getElementType(), loc, in reshapeForBatchMatmul()
340 out = createReshapeOp(out, reshape_shape, original_type.getElementType(), in rewriteToBatchMatmul()
|
| D | unroll_batch_matmul.h | 36 static TF::ReshapeOp createReshapeOp(Value value, ArrayRef<int64_t> shape,
|