| /external/armnn/src/armnn/test/optimizations/ |
| D | PermuteDepthwiseConv2dWeightsTests.cpp | 70 TensorShape newShape = permuteLayer->GetOutputSlot().GetTensorInfo().GetShape(); in PermuteDepthwiseConv2dWeightsTestRunner() local
71 CHECK((newShape[0] == weightsShape[0])); in PermuteDepthwiseConv2dWeightsTestRunner()
72 CHECK((newShape[1] == weightsShape[3])); in PermuteDepthwiseConv2dWeightsTestRunner()
73 CHECK((newShape[2] == weightsShape[1])); in PermuteDepthwiseConv2dWeightsTestRunner()
74 CHECK((newShape[3] == weightsShape[2])); in PermuteDepthwiseConv2dWeightsTestRunner()
|
| /external/armnn/src/armnnUtils/ |
| D | TensorUtils.cpp | 112 std::vector<unsigned int> newShape; in ReduceDims() local
125 newShape.push_back(tensorShape[i]); in ReduceDims()
129 return TensorShape(static_cast<unsigned int>(newShape.size()), newShape.data()); in ReduceDims()
176 std::vector<unsigned int> newShape; in ExpandDimsToRank() local
181 newShape.push_back(1); in ExpandDimsToRank()
187 newShape.push_back(tensorShape[i]); in ExpandDimsToRank()
190 return TensorShape(static_cast<unsigned int>(newShape.size()), newShape.data()); in ExpandDimsToRank()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_shape_ops_to_standard.cc | 64 auto newShape = targetShapeType.getElementType().isIndex() in matchAndRewrite() local
69 rewriter.create<shape::RankOp>(loc, indexType, newShape); in matchAndRewrite()
75 rewriter.create<shape::ReduceOp>(loc, newShape, negOne); in matchAndRewrite()
95 Value extent = b.create<shape::GetExtentOp>(loc, indexType, newShape, in matchAndRewrite()
131 auto newShape = targetShapeType.getElementType().isIndex() in matchAndRewrite() local
136 loc, newShape, llvm::makeArrayRef({one, zero, zero})); in matchAndRewrite()
|
| D | legalize_to_linalg.cc | 2073 llvm::SmallVectorImpl<int64_t>& newShape, in matchAndRewrite()
2075 newShape.reserve(oldShape.size() + 1); in matchAndRewrite()
2078 newShape.push_back(factor); in matchAndRewrite()
2079 newShape.push_back(oldShape[reshapedDim] / factor); in matchAndRewrite()
2081 newShape.push_back(oldShape[i]); in matchAndRewrite()
2097 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local
2098 reshapeShapeVector(prevDimsRef, newShape, inputFeatureDimension, in matchAndRewrite()
2103 RankedTensorType::get(newShape, paddedLhsType.getElementType()), in matchAndRewrite()
2112 llvm::SmallVector<int64_t> newShape; in matchAndRewrite() local
2113 reshapeShapeVector(prevDimsRef, newShape, kernelOutputFeatureDimension, in matchAndRewrite()
[all …]
|
| /external/armnn/src/armnnTfLiteParser/test/ |
| D | Reshape.cpp | 15 const std::string& newShape) in ReshapeFixture()
61 if (!newShape.empty()) in ReshapeFixture()
63 m_JsonString += R"("new_shape" : )" + newShape; in ReshapeFixture()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 221 func (t *Tensor) Reshape(newShape []int64) error {
223 newShapeSize := numElements(newShape)
226 …num_elements: %d) into shape %v (num_elements: %d)", t.shape, oldShapeSize, newShape, newShapeSize)
229 if len(newShape) == 0 {
234 shapePtr = (*C.int64_t)(unsafe.Pointer(&newShape[0]))
237 C.TF_TensorBitcastFrom(t.c, C.TF_TensorType(t.c), t.c, shapePtr, C.int(len(newShape)), status.c)
242 t.shape = newShape
|
| /external/armnn/delegate/common/src/ |
| D | DelegateUtils.hpp | 55 …const armnn::TensorShape& newShape = armnnUtils::ExpandDimsToRank(smallInfo.GetShape(), biggerInpu… in ExpandTensorRankToEqual() local
57 smallInfo.SetShape(newShape); in ExpandTensorRankToEqual()
|
| /external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/ |
| D | NativeInterpreterWrapper.java | 192 int[] newShape = tensor.getInputShapeIfDifferent(input.getValue()); in runSignature() local
193 if (newShape != null) { in runSignature()
194 signatureRunnerWrapper.resizeInput(input.getKey(), newShape); in runSignature() local
234 int[] newShape = tensor.getInputShapeIfDifferent(inputs[i]); in run() local
235 if (newShape != null) { in run()
236 resizeInput(i, newShape); in run()
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeTensorUtils.cpp | 151 std::vector<unsigned int> newShape; in ReduceDimsForACL() local
169 newShape.insert(newShape.begin(), tensorShape[i]); in ReduceDimsForACL()
173 return newShape; in ReduceDimsForACL()
|
| /external/armnn/include/armnn/ |
| D | Tensor.hpp | 193 void SetShape(const TensorShape& newShape) { m_Shape = newShape; } in SetShape() argument
|
| /external/tensorflow/tensorflow/lite/objc/tests/ |
| D | TFLSignatureRunnerTest.m | 103 NSArray<NSNumber *> *newShape = @[ @(2) ];
104 XCTAssertTrue([addRunner resizeInputTensorWithName:@"x" toShape:newShape error:&error]);
113 XCTAssertTrue([[inputTensor shapeWithError:&error] isEqualToArray:newShape]);
|
| /external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| D | armnn_tensor.i | 135 void SetShape(const TensorShape& newShape);
|
| /external/armnn/src/armnnOnnxParser/ |
| D | OnnxParser.cpp | 1066 std::vector<uint32_t> newShape; in To1DTensor() local
1077 newShape.push_back(shape[shape.GetNumDimensions() - 1]); in To1DTensor()
1079 …o[name].m_info->SetShape(TensorShape(static_cast<unsigned int>(newShape.size()), newShape.data())); in To1DTensor()
2346 std::vector<uint32_t> newShape; in PrependForBroadcast() local
2349 newShape.push_back(1); in PrependForBroadcast()
2354 newShape.push_back(input0Shape[dim]); in PrependForBroadcast()
2356 …outputTensorInfo.SetShape(TensorShape(static_cast<unsigned int>(newShape.size()), newShape.data())… in PrependForBroadcast()
|