Searched refs:varTensor (Results 1 – 4 of 4) sorted by relevance
32 MPSGraphTensor* varTensor = getMPSGraphTensor(graphNode->var_id());49 MPSGraphTensor* reshapedVarTensor = [_mpsGraph reshapeTensor:varTensor67 //For now just return meanTensor and varTensor assuming this isn't training70 _idToMPSGraphTensor[graphNode->output2_id()] = [_mpsGraph identityWithTensor:varTensor name:nil];
51 …ConstTensor varTensor(batchNormLayer->m_Variance->GetTensorInfo(), batchNormLayer->m_Variance->Map… in Run() local78 const auto* varBuffer = static_cast<const T*>(varTensor.GetMemoryArea()); in Run()84 std::vector<T> varianceVector(varBuffer, varBuffer + varTensor.GetNumElements()); in Run()
199 MPSGraphTensor* varTensor = nil;205 varTensor = batchVarianceTensor;261 varTensor = saveVarTensor;267 varianceTensor:varTensor
1623 auto varTensor = CreateConstTensor(node.input(4)); in ParseBatchNormalization() local1627 varTensor.first, in ParseBatchNormalization()