Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2259 auto multiples_type = op.multiples().getType().dyn_cast<RankedTensorType>(); in Verify() local
2262 if (multiples_type && multiples_type.getRank() != 1) { in Verify()
2264 << multiples_type.getRank(); in Verify()
2267 if (input_type && multiples_type && multiples_type.hasStaticShape() && in Verify()
2268 (input_type.getRank() != multiples_type.getNumElements() || in Verify()
2269 (input_type.getRank() == 0 && multiples_type.getNumElements() == 1))) { in Verify()
2272 << ", got multiples of size " << multiples_type.getNumElements() in Verify()