Searched refs:expectedShape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/java/src/test/java/org/tensorflow/lite/ |
| D | TensorTest.java | 81 int[] expectedShape = {2, 8, 8, 3}; in testBasic() local
82 assertThat(tensor.shape()).isEqualTo(expectedShape); in testBasic()
83 assertThat(tensor.shapeSignature()).isEqualTo(expectedShape); in testBasic()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 720 llvm::SmallVector<int64_t, 10> expectedShape(sizes.getValues<int64_t>()); in Verify() local
723 expectedShape.insert(expectedShape.end(), operandShape.begin(), in Verify()
727 if (resultShape != llvm::makeArrayRef(expectedShape)) { in Verify()
731 llvm::make_range(expectedShape.begin(), expectedShape.end()))); in Verify()
2714 SmallVector<int64_t, 4> expectedShape(operandRank); in Verify() local
2717 expectedShape[i] = operandType.getDimSize(permutedDim); in Verify()
2721 RankedTensorType::get(expectedShape, resultType.getElementType()); in Verify()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 357 auto expectedShape = llvm::to_vector<4>( in verifyOutputShape() local
362 VectorType::get(expectedShape, resVectorType.getElementType()); in verifyOutputShape()
|