Searched refs:betaTensor (Results 1 – 10 of 10) sorted by relevance
83 ScopedTensorHandle betaTensor(tensorInfo);88 AllocateAndCopyDataToITensorHandle(&betaTensor, beta.data());95 data.m_Beta = &betaTensor;
77 armnn::ScopedTensorHandle betaTensor(tensorInfo); in BatchNormTestImpl() local83 descriptor.m_Beta = &betaTensor; in BatchNormTestImpl()91 AllocateAndCopyDataToITensorHandle(&betaTensor, beta.data()); in BatchNormTestImpl()166 armnn::ScopedTensorHandle betaTensor(tensorInfo); in BatchNormTestNhwcImpl() local171 AllocateAndCopyDataToITensorHandle(&betaTensor, &beta[0]); in BatchNormTestNhwcImpl()178 data.m_Beta = &betaTensor; in BatchNormTestNhwcImpl()656 armnn::ScopedTensorHandle betaTensor(tensorInfo); in CompareBatchNormTest() local661 AllocateAndCopyDataToITensorHandle(&betaTensor, &beta[0]); in CompareBatchNormTest()668 data.m_Beta = &betaTensor; in CompareBatchNormTest()
68 MPSGraphTensor* betaTensor = [_mpsGraph constantWithScalar:beta72 secondaryTensor:betaTensor
63 betaTensor:reshapedBiasTensor115 betaTensor:biasTensor
48 …ConstTensor betaTensor(batchNormLayer->m_Beta->GetTensorInfo(), batchNormLayer->m_Beta->Map(true)); in Run() local75 const auto* betaBuffer = static_cast<const T*>(betaTensor.GetMemoryArea()); in Run()81 std::vector<T> betaVector (betaBuffer, betaBuffer + betaTensor.GetNumElements()); in Run()
156 MPSGraphTensor* betaTensor = [mpsGraph constantWithScalar:beta_.toDouble()160 secondaryTensor:betaTensor
369 MPSGraphTensor* betaTensor = [mpsGraph constantWithScalar:beta.toDouble()389 secondaryTensor:betaTensor480 auto betaTensor = [mpsGraph constantWithScalar:beta.toDouble()483 secondaryTensor:betaTensor880 MPSGraphTensor* betaTensor = [mpsGraph constantWithScalar:beta.toDouble()892 secondaryTensor:betaTensor
616 MPSGraphTensor* betaTensor = [mpsGraph constantWithScalar:beta dataType:inputTensor.dataType];632 secondaryTensor:betaTensor645 secondaryTensor:betaTensor748 MPSGraphTensor* betaTensor = [mpsGraph constantWithScalar:beta dataType:MPSDataTypeFloat32];757 secondaryTensor:betaTensor761 secondaryTensor:betaTensor
1152 MPSGraphTensor* betaTensor = mpsGraphScalarPlaceHolder(mpsGraph, inputTensor.dataType);1158 MPSGraphTensor* reciprocalBetaTensor = [mpsGraph reciprocalWithTensor:betaTensor name:nil];1160 secondaryTensor:betaTensor1176 newCachedGraph->betaTensor_ = betaTensor;1225 MPSGraphTensor* betaTensor = mpsGraphScalarPlaceHolder(mpsGraph, inputTensor.dataType);1231 secondaryTensor:betaTensor1251 newCachedGraph->betaTensor_ = betaTensor;
269 betaTensor:biasTensor