- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for VerifyShapeOfReshapeOp (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
rhs_reshape_segids.at(std::get<0>(i)) = num_rhs_reshape_segids; } ++num_rhs_reshape_segids; // If LHS requires reshapes. if (lhs_rank != num_lhs_reshape_segids) { if (succeeded(VerifyShapeOfReshapeOp(lhs_shape))) { *lhs = createReshapeOp(*lhs, lhs_shape, lhs_type.getElementType(), loc, rewriter); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc
if (i == pack_axis) { pack_out_shape.push_back(count); } else { pack_out_shape.push_back(input_type.getDimSize(j)); j++; } } if (failed(TF::VerifyShapeOfReshapeOp(pack_out_shape))) return failure(); // Insert the concat op. auto concat_out_type = RankedTensorType::get(concat_out_shape, input_type.getElementType());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 25.4K bytes - Viewed (0)