Searched defs:CheckShape (Results 1 – 5 of 5) sorted by relevance
230 bool CheckShape(bool verbose, const TfLiteTensor& tensor) { in CheckShape() function in tflite::testing::TfLiteDriver::ShapeExpectation
1700 Status ShapeVerifier::CheckShape(const HloInstruction* instruction, in CheckShape() function in xla::ShapeVerifier1772 Status ShapeVerifier::CheckShape(const HloInstruction* instruction, in CheckShape() function in xla::ShapeVerifier
69 XlaOp CheckShape(XlaBuilder* b, const XlaOp operand, in CheckShape() function in xla::__anon51499d460111::BatchNormalizationTest
627 bool CheckShape(const armnn::TensorShape& actual, const std::vector<int32_t>& expected) in CheckShape() function647 bool CheckShape(const armnn::TensorShape& actual, const armnn::TensorShape& expected) in CheckShape() function
188 bool CheckShape(const armnn::TensorShape& actual, const std::vector<uint32_t>& expected) in CheckShape() function