- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for VerifyShapeOfReshapeOp (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/verification_utils.cc
#include "mlir/IR/BuiltinTypes.h" // from @llvm-project #include "mlir/IR/Types.h" // from @llvm-project #include "mlir/Support/LogicalResult.h" // from @llvm-project namespace mlir { namespace TF { LogicalResult VerifyShapeOfReshapeOp(ArrayRef<int64_t> shape) { bool has_dynamic_dim = false; for (int64_t dim : shape) { if (dim != ShapedType::kDynamic) { if (dim < 0) return failure(); continue; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Nov 21 16:21:18 UTC 2022 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize.cc
new_reshape_dim_idx++; } } new_reshape_dims.push_back(new_reshape_dim); } if (non_unit_dims != old_reshape_non_unit_dims) return failure(); if (failed(VerifyShapeOfReshapeOp(new_reshape_dims))) return failure(); Type el_ty = getElementTypeOrSelf(op.getType()); TF::ConstOp new_reshape_shape = GetI64ConstantTensor(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0)