Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc299 auto crops_type = op.crops().getType().cast<TensorType>(); in Verify() local
300 if (crops_type.hasRank()) { in Verify()
301 if (crops_type.getRank() != 2) in Verify()
303 << "requires crops to be a 2D tensor, but got " << crops_type; in Verify()
306 if (crops_type.isDynamicDim(dim)) return true; in Verify()
307 return crops_type.getDimSize(dim) == size; in Verify()
311 << "requires crops to be a tensor<2x2>, but got " << crops_type; in Verify()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_common.cc1056 RankedTensorType crops_type = in convertBatchToSpaceNDOp() local
1071 if (!crops_type) { in convertBatchToSpaceNDOp()
1079 int crops_dims = crops_type.getShape()[0]; in convertBatchToSpaceNDOp()