PrimitiveFunction.cpp
//
// Copyright (c) Microsoft. All rights reserved.
// Licensed under the MIT license. See LICENSE.md file in the project root for full license information.
//
#include "stdafx.h"
#include "PrimitiveFunction.h"
#include "ComputationNode.h"
#include "ReshapingNodes.h"
#include "EvaluationNodes.h"
#include "TrainingNodes.h"
#include "LinearAlgebraNodes.h"
#include "InputAndParamNodes.h"
#include "NonlinearityNodes.h"
#include "RecurrentNodes.h"
#include "Serialization.h"
#include "RNNNodes.h"
#include "BlockFunction.h"
#include "CompositeFunction.h"
#include "SpecialPurposeNodes.h"
#include "ConvolveGeometry.h"
#include "ConvolutionalNodes.h"
#include "Variable.h"
using namespace Microsoft::MSR::CNTK;
namespace CNTK
{
// Names for the reduction operations as used by the CNTK ReduceElementsNode
/*static*/ const std::wstring PrimitiveFunction::InternalSumReductionOpName = L"Sum";
/*static*/ const std::wstring PrimitiveFunction::InternalLogSumReductionOpName = L"LogSum";
/*static*/ const std::wstring PrimitiveFunction::InternalMeanReductionOpName = L"Mean";
/*static*/ const std::wstring PrimitiveFunction::InternalMaxReductionOpName = L"Max";
/*static*/ const std::wstring PrimitiveFunction::InternalMinReductionOpName = L"Min";
/*static*/ const std::wstring PrimitiveFunction::InternalProdReductionOpName = L"Prod";
/*static*/ const std::wstring PrimitiveFunction::InternalAllReductionOpName = L"All";
/*static*/ const std::wstring PrimitiveFunction::InternalAnyReductionOpName = L"Any";
/*static*/ const std::wstring PrimitiveFunction::InternalArgmaxReductionOpName = L"Argmax";
/*static*/ const std::wstring PrimitiveFunction::InternalArgminReductionOpName = L"Argmin";
// Names of the various attributes of CNTK primitive Functions
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAxis = L"axis";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAxisVec = L"axisVec";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAxis1 = L"axis1";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAxis2 = L"axis2";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAllowDuplicates = L"allowDuplicates";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNumSamples = L"numSamples";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameDropoutRate = L"dropoutRate";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNewShape = L"newShape";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBeginAxis = L"beginAxis";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameEndAxis = L"endAxis";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameOutputRank = L"outputRank";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameInferInputRankToMap = L"inferInputRankToMap";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameOffset = L"offset";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameStrides = L"strides";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameSharing = L"sharing";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameAutoPadding = L"autoPadding";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameLowerPad = L"lowerPad";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameUpperPad = L"upperPad";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameCeilOutDim = L"ceilOutDim";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameIncludePad = L"includePad";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameTranspose = L"transpose";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameOutputShape = L"outputShape";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameMaxTempMemSizeInSamples = L"maxTempMemSizeInSamples";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameROIOutputShape = L"roiOutputShape";
/*static*/ const std::wstring PrimitiveFunction::AttributeNamePoolingType = L"poolingType";
/*static*/ const std::wstring PrimitiveFunction::AttributeNamePoolingWindowShape = L"poolingWindowShape";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameSpatial = L"spatial";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNormalizationTimeConstant = L"normalizationTimeConstant";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBlendTimeConstant = L"blendTimeConstant";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameEpsilon = L"epsilon";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameUseCuDNNEngine = L"useCuDNNEngine";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNewDynamicAxes = L"newDynamicAxes";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNewSequenceAxisLengthScalingFactor = L"newSequenceAxisLengthScalingFactor";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNewSequenceAxisLengthAdditiveFactor = L"newSequenceAxisLengthAdditiveFactor";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBeginIndex = L"beginIndex";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBeginIndexVec = L"beginIndexVec";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameEndIndex = L"endIndex";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameEndIndexVec = L"endIndexVec";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameReductionOpName = L"reductionOpName";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBidirectional = L"bidirectional";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNumLayers = L"numLayers";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameHiddenSize = L"hiddenSize";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameRecurrentOp = L"recurrentOp";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameRngSeed = L"rngSeed";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameRngOffset = L"rngOffset";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameUnpoolingWindowShape = L"unpoolingWindowShape";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameSubstitutionPenalty = L"SubstitutionPenalty";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameDeletionPenalty = L"DeletionPenalty";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameInsertionPenalty = L"InsertionPenalty";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameSquashInputs = L"SquashInputs";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameTokensToIgnore = L"TokensToIgnore";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameDelayConstraint = L"DelayConstraint";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameBlankTokenId = L"BlankTokenId";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameNumClass = L"numClass";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameOneHotOutputSparse = L"oneHotOutputSparse";
/*static*/ const std::wstring PrimitiveFunction::AttributeNameOneHotAxis = L"onehotAxis";
/*static*/ DataType PrimitiveFunction::GetOutputDataType(PrimitiveOpType op, std::vector<Variable>& inputs, bool inferDimensions)
{
// We use the first non-constant input operand's DataType as the output DataType
// In case there are no non-constant known DataTypes, we just pick the first known operand DataType
// Also, all the known DataTypes of operands should match except for constants where coercion is allowed
DataType firstKnownInputDataType = DataType::Unknown;
DataType outputDataType = DataType::Unknown;
size_t i = 0;
while (i < inputs.size())
{
auto input = inputs[i++];
auto inputDataType = input.GetDataType();
if (inputDataType != DataType::Unknown)
{
if (firstKnownInputDataType == DataType::Unknown)
firstKnownInputDataType = inputDataType;
if (outputDataType == DataType::Unknown)
{
if (!input.IsConstant())
outputDataType = inputDataType;
}
else
{
// The DataType of all operands should match except for Constants where we allow coercion
if ((inputDataType != DataType::Unknown) && (inputDataType != outputDataType) && !input.IsConstant())
InvalidArgument("Primitive op '%S' passed operands '%S' with different DataTypes '%s' and '%s'.",
PrimitiveOpTypeName(op).c_str(), NamedListString(inputs).c_str(), DataTypeName(outputDataType), DataTypeName(inputDataType));
}
}
}
if (outputDataType == DataType::Unknown)
outputDataType = firstKnownInputDataType;
// Propagate the data type to any input Parameters/Constants with unknown data type
if (inferDimensions && (outputDataType != DataType::Unknown))
{
for (auto& input : inputs)
{
if ((input.GetDataType() == DataType::Unknown) && (input.IsConstant() || input.IsParameter()))
input.m_dataFields->m_dataType = outputDataType;
}
}
return outputDataType;
}
/*static*/ std::vector<Axis> PrimitiveFunction::GetOutputDynamicAxes(PrimitiveOpType op, std::vector<Variable>& inputs, PrimitiveFunction* owner, Dictionary& functionConfig)
{
auto reduceAxis = [](Axis reductionAxis, Variable input, std::vector<Axis>& outputDynamicAxes)
{
reductionAxis = NormalizeAxis(reductionAxis, input);
for (auto inputDynamicAxis : input.DynamicAxes())
{
if (inputDynamicAxis != reductionAxis)
outputDynamicAxes.push_back(inputDynamicAxis);
}
};
// We currently require that the inputs' dynamic axes, if any, match
std::vector<Axis> outputDynamicAxes;
if ((op == PrimitiveOpType::SumAll) ||
(op == PrimitiveOpType::ReduceElements && functionConfig[PrimitiveFunction::AttributeNameAxis].Value<Axis>() == Axis::AllAxes()) ||
(op == PrimitiveOpType::SquaredError) ||
(op == PrimitiveOpType::CrossEntropyWithSoftmax) ||
(op == PrimitiveOpType::EditDistanceError) ||
(op == PrimitiveOpType::ClassificationError) ||
(op == PrimitiveOpType::Logistic) ||
(op == PrimitiveOpType::LambdaRank) ||
(op == PrimitiveOpType::NDCG))
{
outputDynamicAxes = std::vector<Axis>({});
}
else if ((op == PrimitiveOpType::ReduceElements) && functionConfig[PrimitiveFunction::AttributeNameAxis].Value<Axis>().IsDynamicAxis() && (inputs[0].DynamicAxes() != Axis::UnknownDynamicAxes()))
{
reduceAxis(functionConfig[PrimitiveFunction::AttributeNameAxis].Value<Axis>(), inputs[0], outputDynamicAxes);
}
else if ((op == PrimitiveOpType::Times) && (functionConfig[PrimitiveFunction::AttributeNameInferInputRankToMap].Value<int>() == TimesReduceSequenceAxisWithoutInferredInputRank))
{
reduceAxis(Axis::OperandSequenceAxis(), inputs[0], outputDynamicAxes);
}
else if (op == PrimitiveOpType::Where)
{
if (functionConfig.Contains(PrimitiveFunction::AttributeNameNewDynamicAxes))
outputDynamicAxes = AsVector<Axis>(functionConfig[PrimitiveFunction::AttributeNameNewDynamicAxes].Value<std::vector<DictionaryValue>>());
else
{
if (inputs[0].DynamicAxes() == Axis::UnknownDynamicAxes())
outputDynamicAxes = Axis::UnknownDynamicAxes();
else
{
if (functionConfig.Contains(PrimitiveFunction::AttributeNameNewSequenceAxisLengthScalingFactor) &&
functionConfig.Contains(PrimitiveFunction::AttributeNameNewSequenceAxisLengthAdditiveFactor))
{
size_t newSequenceAxisLengthScalingFactor = functionConfig[PrimitiveFunction::AttributeNameNewSequenceAxisLengthScalingFactor].Value<size_t>();
int newSequenceAxisLengthAdditiveFactor = functionConfig[PrimitiveFunction::AttributeNameNewSequenceAxisLengthAdditiveFactor].Value<int>();
auto derivedDynamicAxes = GetDerivedDynamicAxes(inputs[0].DynamicAxes()[0], newSequenceAxisLengthScalingFactor, newSequenceAxisLengthAdditiveFactor);
std::copy(derivedDynamicAxes.begin(), derivedDynamicAxes.end(), std::back_inserter(outputDynamicAxes));
}
else
{
std::function<Variable(const Variable&)> GetActualSourceVariable;
GetActualSourceVariable = [&GetActualSourceVariable](const Variable& var) -> Variable {
if (var.BlockFunctionVariableMapping() == Variable())
return var;
else
return GetActualSourceVariable(var.BlockFunctionVariableMapping());
};
auto whereNodeConditionSourceVar = GetActualSourceVariable(inputs[0]);
auto whereNodeSequenceAxis = Axis(std::wstring(L"whereNodeDynamicAxis_conditionVar_") + whereNodeConditionSourceVar.Uid());
outputDynamicAxes.push_back(whereNodeSequenceAxis);
}
for (size_t i2 = 1; i2 < inputs[0].DynamicAxes().size(); ++i2)
outputDynamicAxes.push_back(inputs[0].DynamicAxes()[i2]);
functionConfig[PrimitiveFunction::AttributeNameNewDynamicAxes] = AsDictionaryValueVector(outputDynamicAxes);
}
}
}
else if (op == PrimitiveOpType::ScatterPacked)
outputDynamicAxes = inputs[2].DynamicAxes();
else if ((op == PrimitiveOpType::PackedIndex) || (op == PrimitiveOpType::GatherPacked))
outputDynamicAxes = inputs[1].DynamicAxes();
else if (op == PrimitiveOpType::ReconcileDynamicAxis)
outputDynamicAxes = inputs[1].DynamicAxes();
else if (op == PrimitiveOpType::PastValue || op == PrimitiveOpType::FutureValue)
outputDynamicAxes = inputs[0].DynamicAxes(); // second arg (initial state) may have different dynamic axis
else
{
auto allInputDynamicAxesEmpty = std::find_if(inputs.begin(), inputs.end(), [](const Variable& input) { return !input.DynamicAxes().empty(); }) == inputs.end();
if (!allInputDynamicAxesEmpty)
{
outputDynamicAxes = Axis::UnknownDynamicAxes();
for (auto inputVar : inputs)
{
auto currentInputDynamicAxes = inputVar.DynamicAxes();
if (!currentInputDynamicAxes.empty() && (currentInputDynamicAxes != Axis::UnknownDynamicAxes()))
{
if (outputDynamicAxes == Axis::UnknownDynamicAxes())
outputDynamicAxes = currentInputDynamicAxes;
else
{
if (currentInputDynamicAxes != outputDynamicAxes)
LogicError("Operation '%S': Operand '%S' has dynamic axes, that do not match the dynamic axes '%S' of the other operands.",
PrimitiveOpTypeName(op).c_str(), inputVar.AsString().c_str(), DynamicAxesAsString(outputDynamicAxes).c_str());
}
}
}
}
}
return outputDynamicAxes;
}
void PrimitiveFunction::InferOutputs(std::vector<Variable>& outputs)
{
if (m_op == PrimitiveOpType::Combine)
outputs.assign(m_inputs.begin(), m_inputs.end());
else if (m_op == PrimitiveOpType::NoOp)
outputs.push_back(OutputVariable(m_inputs[0].Shape(), m_inputs[0].GetDataType(), m_inputs[0].DynamicAxes(), m_inputs[0].NeedsGradient(), Name()));
else
{
DataType outputDataType = GetOutputDataType(m_op, m_inputs, true);
std::vector<Axis> outputDynamicAxes = GetOutputDynamicAxes(m_op, m_inputs, this, m_attributes);
bool needsGradient = std::any_of(m_inputs.begin(), m_inputs.end(), [](const Variable& input) { return input.NeedsGradient(); });
NDShape outputShape = NDShape::Unknown;
bool allInputShapesUnknown = (std::find_if(m_inputs.begin(), m_inputs.end(), [](const Variable& input) { return !input.Shape().IsUnknown(); }) == m_inputs.end());
bool anyInputShapesUnknown = (std::find_if(m_inputs.begin(), m_inputs.end(), [](const Variable& input) { return input.Shape().IsUnknown(); }) != m_inputs.end());
if (!anyInputShapesUnknown || (!allInputShapesUnknown && (outputDynamicAxes != Axis::UnknownDynamicAxes())))
{
switch (m_op)
{
// Elementwise operators' shapes are a zip of inputs and can be determined even if some of the input shapes are unknown
case PrimitiveOpType::Plus:
case PrimitiveOpType::LogPlus:
case PrimitiveOpType::Pow:
case PrimitiveOpType::Minus:
case PrimitiveOpType::ElementTimes:
case PrimitiveOpType::Equal:
case PrimitiveOpType::NotEqual:
case PrimitiveOpType::Less:
case PrimitiveOpType::LessEqual:
case PrimitiveOpType::Greater:
case PrimitiveOpType::GreaterEqual:
case PrimitiveOpType::PastValue:
case PrimitiveOpType::FutureValue:
{
assert(m_inputs.size() == 2);
if ((m_op == PrimitiveOpType::PastValue) || (m_op == PrimitiveOpType::FutureValue))
{
Variable inputOperandVar = m_inputs[0];
Variable initialStateVar = m_inputs[1];
// TODO: We currently only support input operand with 1 dynamic axis for PastValue/FutureValue
if ((inputOperandVar.DynamicAxes() != Axis::UnknownDynamicAxes()) && (inputOperandVar.DynamicAxes().size() != 2))
LogicError("PastValue/FutureValue Function '%S': Input operand '%S' with #dynamic axes != 2 (1 sequence axis and 1 batch axis) is not supported.", AsString().c_str(), inputOperandVar.AsString().c_str());
}
outputShape = BinaryElementwiseOpOutputShape(m_op, m_inputs[0], m_inputs[1], true, true);
break;
}
case PrimitiveOpType::Clip:
assert(m_inputs.size() == 3);
outputShape = NaryElementwiseOpOutputShape(m_op, m_inputs, true, true);
break;
case PrimitiveOpType::Select:
assert(m_inputs.size() == 3);
outputShape = NaryElementwiseOpOutputShape(m_op, m_inputs, true, true);
break;
default:
// For all other operations, shapes of all inputs must be known to determine the output shape
if (!anyInputShapesUnknown)
{
switch (m_op)
{
case PrimitiveOpType::Negate:
case PrimitiveOpType::Sigmoid:
case PrimitiveOpType::Tanh:
case PrimitiveOpType::ReLU:
case PrimitiveOpType::Exp:
case PrimitiveOpType::Log:
case PrimitiveOpType::Sqrt:
case PrimitiveOpType::Floor:
case PrimitiveOpType::Abs:
case PrimitiveOpType::Reciprocal:
case PrimitiveOpType::Softmax:
case PrimitiveOpType::Hardmax:
case PrimitiveOpType::Dropout:
case PrimitiveOpType::LogSoftmax:
case PrimitiveOpType::Sin:
case PrimitiveOpType::Cos:
case PrimitiveOpType::Pass:
case PrimitiveOpType::LabelsToGraph:
case PrimitiveOpType::StopGradient:
case PrimitiveOpType::ELU:
assert(m_inputs.size() == 1);
outputShape = UnaryElementwiseOpOutputShape(m_inputs[0].Shape());
break;
case PrimitiveOpType::Where:
assert(m_inputs.size() == 1);
outputShape = NDShape{}; // scalar
break;
case PrimitiveOpType::PackedIndex:
assert(m_inputs.size() == 2);
outputShape = UnaryElementwiseOpOutputShape(m_inputs[1].Shape());
break;
case PrimitiveOpType::ScatterPacked:
{
assert(m_inputs.size() == 3);
if (m_inputs[0].DynamicAxes().empty() || m_inputs[1].DynamicAxes().empty() || m_inputs[2].DynamicAxes().empty())
InvalidArgument("ScatterPacked: All operands '%S' must have dynamic axes.", NamedListString(m_inputs).c_str());
outputShape = UnaryElementwiseOpOutputShape(m_inputs[0].Shape());
break;
}
case PrimitiveOpType::TransposeAxes:
{
assert(m_inputs.size() == 1);
auto axis1 = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameAxis1].Value<Axis>(), m_inputs[0].Shape());
auto axis2 = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameAxis2].Value<Axis>(), m_inputs[0].Shape());
if (!axis1.IsStaticAxis() || !axis2.IsStaticAxis())
LogicError("Function '%S': TransposeAxes operation currently does not support transposing dynamic axes.", AsString().c_str());
// We allow to transpose with an axes that exceeds the rank of the input.
// The output rank is the max of the input rank, and either of the axes being transposed.
auto outputRank = std::max(m_inputs[0].Shape().Rank(), (size_t)(std::max(axis1.StaticAxisIndex(), axis2.StaticAxisIndex()) + 1));
outputShape = m_inputs[0].Shape().AppendShape(NDShape(outputRank - m_inputs[0].Shape().Rank(), 1));
std::swap(outputShape[axis1.StaticAxisIndex()], outputShape[axis2.StaticAxisIndex()]);
break;
}
case PrimitiveOpType::Slice:
{
assert(m_inputs.size() == 1);
std::vector<Axis> axis;
std::vector<int> beginIndex, endIndex;
if (m_attributes.Contains(PrimitiveFunction::AttributeNameAxisVec) &&
m_attributes.Contains(PrimitiveFunction::AttributeNameBeginIndexVec) &&
m_attributes.Contains(PrimitiveFunction::AttributeNameEndIndexVec))
{
auto &axisDictionary = m_attributes[PrimitiveFunction::AttributeNameAxisVec].Value<std::vector<DictionaryValue>>();
for (auto& value : axisDictionary)
axis.push_back(NormalizeStaticAxis(value.Value<Axis>(), m_inputs[0].Shape()));
beginIndex = AsVector<int>(m_attributes[PrimitiveFunction::AttributeNameBeginIndexVec].Value<std::vector<DictionaryValue>>());
endIndex = AsVector<int>(m_attributes[PrimitiveFunction::AttributeNameEndIndexVec].Value<std::vector<DictionaryValue>>());
}
else if (m_attributes.Contains(PrimitiveFunction::AttributeNameAxis) &&
m_attributes.Contains(PrimitiveFunction::AttributeNameBeginIndex) &&
m_attributes.Contains(PrimitiveFunction::AttributeNameEndIndex))
{
axis.push_back(NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameAxis].Value<Axis>(), m_inputs[0].Shape()));
beginIndex.push_back(m_attributes[PrimitiveFunction::AttributeNameBeginIndex].Value<int>());
endIndex.push_back(m_attributes[PrimitiveFunction::AttributeNameEndIndex].Value<int>());
}
else
{
RuntimeError("Function '%S': Slice operation with inconsistent attributes", AsString().c_str());
}
auto outputTensorShape = AsTensorShape(m_inputs[0].Shape());
for (auto i = 0; i < axis.size(); i++)
{
auto& ax = axis[i];
if (!ax.IsStaticAxis())
LogicError("Function '%S': Built-in Slice operation currently does not support slicing along dynamic axis.", AsString().c_str());
VerifyStaticAxis(ax, m_inputs[0].Shape());
size_t sliceAxisDim = m_inputs[0].Shape()[ax.StaticAxisIndex()];
int realBeginIndex = (beginIndex[i] >= 0) ? beginIndex[i] : beginIndex[i] + sliceAxisDim;
int realEndIndex = (endIndex[i] > 0) ? endIndex[i] : endIndex[i] + sliceAxisDim;
if ((sliceAxisDim < realEndIndex) || (realEndIndex < realBeginIndex) || (realBeginIndex < 0))
RuntimeError("Function '%S': Slice operation index range [%d,%d), interpreted as [%d,%d), is invalid for input '%S' shape '%S'.",
AsString().c_str(),
beginIndex[i],
endIndex[i],
realBeginIndex,
realEndIndex,
m_inputs[0].AsString().c_str(),
m_inputs[0].Shape().AsString().c_str());
// propagate as much as we can
if ((ax.StaticAxisIndex() < (int)outputTensorShape.GetRank()) && (0 <= realBeginIndex) && (realBeginIndex <= realEndIndex) && (realEndIndex <= sliceAxisDim))
outputTensorShape.NarrowTo(ax.StaticAxisIndex(), realBeginIndex, realEndIndex);
}
outputShape = AsNDShape(outputTensorShape, /*allowNonFlattenableTensorShapes = */ true);
break;
}
case PrimitiveOpType::Reshape:
{
auto& replacementShape = m_attributes[PrimitiveFunction::AttributeNameNewShape].Value<NDShape>();
auto beginAxis = Axis(0);
auto endAxis = Axis((int)m_inputs[0].Shape().Rank());
if (m_attributes.Contains(PrimitiveFunction::AttributeNameBeginAxis))
beginAxis = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameBeginAxis].Value<Axis>(), m_inputs[0].Shape());
if (m_attributes.Contains(PrimitiveFunction::AttributeNameEndAxis))
endAxis = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameEndAxis].Value<Axis>(), m_inputs[0].Shape());
outputShape = ReshapeOutputShape(m_inputs[0].Shape(), replacementShape, beginAxis, endAxis, true);
break;
}
case PrimitiveOpType::ROIPooling:
{
assert(m_inputs.size() == 2);
auto convMapShape = m_inputs[0].Shape();
auto roisShape = m_inputs[1].Shape();
auto roiOutputShape = m_attributes[PrimitiveFunction::AttributeNameROIOutputShape].Value<NDShape>();
auto outW = roiOutputShape[0];
auto outH = roiOutputShape[1];
auto numChannels = convMapShape[2];
auto roisPerImage = roisShape[1];
if (roiOutputShape.Rank() != 2)
InvalidArgument("ROIPoolingNode: ROI shape '%S' must have rank 2 ([W x H]).", roiOutputShape.AsString().c_str());
if (convMapShape[0] < outW || convMapShape[1] < outH)
InvalidArgument("ROIPoolingNode: input Width (%d) must be >= ROI window Width (%d) and input Height (%d) must be >= ROI window Height (%d).",
(int)convMapShape[0], (int)outW, (int)convMapShape[1], (int)outH);
if (convMapShape[2] < 1)
InvalidArgument("ROIPoolingNode: input '%S' must have at least one channel ([W x H x C]).", m_inputs[0].AsString().c_str());
if (roisShape[0] != 4)
InvalidArgument("ROIPoolingNode: ROI shape '%S' must be of the form: [4 x roisPerImage].", roisShape.AsString().c_str());
if (roisPerImage < 1)
InvalidArgument("ROIPoolingNode: ROI shape '%S' must contain at least one ROI ([4 x roisPerImage]).", roisShape.AsString().c_str());
outputShape = { outW, outH, numChannels, roisPerImage };
break;
}
case PrimitiveOpType::Pooling:
{
assert(m_inputs.size() == 1);
auto poolingWindowsShape = m_attributes[PrimitiveFunction::AttributeNamePoolingWindowShape].Value<NDShape>();
auto strides = m_attributes[PrimitiveFunction::AttributeNameStrides].Value<NDShape>();
auto lowerPad = m_attributes[PrimitiveFunction::AttributeNameLowerPad].Value<NDShape>();
auto upperPad = m_attributes[PrimitiveFunction::AttributeNameUpperPad].Value<NDShape>();
auto autoPadding = AsVector<bool>(m_attributes[PrimitiveFunction::AttributeNameAutoPadding].Value<std::vector<DictionaryValue>>());
bool ceilOutDim = false;
if (m_attributes.Contains(PrimitiveFunction::AttributeNameCeilOutDim))
ceilOutDim = m_attributes[PrimitiveFunction::AttributeNameCeilOutDim].Value<bool>();
NDShape outputMapCount = { 1 };
std::vector<bool> sharing = { true };
auto inputShape = m_inputs[0].Shape();
// In case of pooling if the kernel shape is unknown, then treat it as global pooling.
if (poolingWindowsShape == NDShape::Unknown)
{
if ((std::find(autoPadding.begin(), autoPadding.end(), true) != autoPadding.end()) || (lowerPad.TotalSize() > 0) || (upperPad.TotalSize() > 0))
RuntimeError("Padding isn't allowed for Unknown pooling window shape!");
poolingWindowsShape = inputShape.SubShape(0, inputShape.Rank() - 1);
m_attributes[PrimitiveFunction::AttributeNamePoolingWindowShape] = poolingWindowsShape;
}
outputShape = ConvolutionOpOutputShape(m_op, inputShape, poolingWindowsShape, outputMapCount, strides, sharing, autoPadding, lowerPad, upperPad, false, true, ceilOutDim);
break;
}
case PrimitiveOpType::Unpooling:
{
assert(m_inputs.size() == 2);
auto inputShape = m_inputs[0].Shape();
outputShape = m_inputs[1].Shape();
PoolingType unpoolingType = (PoolingType)(m_attributes[PrimitiveFunction::AttributeNamePoolingType].Value<size_t>());
if (unpoolingType != PoolingType::Max)
LogicError("Function '%S': Currently only max unpooling is supported.", AsString().c_str());
// Finding the shape of an unpooling operation from the input to be unpooled alone is ambiguous
// For example a 4x4 input with a 5x5 kernel a stride of 2x2
// and padding could have resulted from pooling a 7x7 or 8x8 image
// Therefore what needs to happen here is to check whether the
// outputShape can be pooled into the inputShape using the specified attributes
auto unpoolingWindowShape = m_attributes[PrimitiveFunction::AttributeNameUnpoolingWindowShape].Value<NDShape>();
auto strides = m_attributes[PrimitiveFunction::AttributeNameStrides].Value<NDShape>();
auto lowerPad = m_attributes[PrimitiveFunction::AttributeNameLowerPad].Value<NDShape>();
auto upperPad = m_attributes[PrimitiveFunction::AttributeNameUpperPad].Value<NDShape>();
auto autoPadding = AsVector<bool>(m_attributes[PrimitiveFunction::AttributeNameAutoPadding].Value<std::vector<DictionaryValue>>());
NDShape inputMapCount = { 1 };
std::vector<bool> sharing = { true };
NDShape inferredInputShape = ConvolutionOpOutputShape(PrimitiveOpType::Pooling, outputShape, unpoolingWindowShape, inputMapCount, strides, sharing, autoPadding, lowerPad, upperPad, false, true);
if (inferredInputShape != inputShape)
RuntimeError("Unpooling: The shape '%S' of the unpooling operand '%S' is different than the shape '%S from pooling the input argument '%S' using the provided options.",
inputShape.AsString().c_str(), m_inputs[0].AsString().c_str(), inferredInputShape.AsString().c_str(), m_inputs[1].AsString().c_str());
break;
}
case PrimitiveOpType::SumAll:
assert(m_inputs.size() == 1);
outputShape = { 1 };
break;
case PrimitiveOpType::OneHot:
{
assert(m_inputs.size() == 1);
auto num_class = m_attributes[PrimitiveFunction::AttributeNameNumClass].Value<size_t>();
auto inputShape = m_inputs[0].Shape();
auto fakeShape = inputShape.AppendShape({num_class});
auto axis = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameOneHotAxis].Value<Axis>(), fakeShape);
if (!axis.IsStaticAxis())
LogicError("Function '%S': one hot operation currently does not support on dynamic axis", AsString().c_str());
size_t len = inputShape.Rank();
int axisIndex = axis.StaticAxisIndex();
outputShape = {};
if (axisIndex > 0)
{
outputShape = outputShape.AppendShape(inputShape.SubShape(0, axisIndex));
}
outputShape = outputShape.AppendShape({num_class});
if (axisIndex < len)
{
outputShape = outputShape.AppendShape(inputShape.SubShape(axisIndex, len));
}
break;
}
case PrimitiveOpType::GatherOp:
{
assert(m_inputs.size() == 2);
auto inputShape1 = m_inputs[0].Shape();
auto inputShape2 = m_inputs[1].Shape();
auto inputDim2 = inputShape2.Dimensions();
inputDim2.pop_back();
outputShape = NDShape(inputDim2);
outputShape = outputShape.AppendShape(inputShape1);
break;
}
case PrimitiveOpType::Times:
{
assert(m_inputs.size() == 2);
auto outputRank = m_attributes[PrimitiveFunction::AttributeNameOutputRank].Value<size_t>();
auto inferInputRankToMap = m_attributes[PrimitiveFunction::AttributeNameInferInputRankToMap].Value<int>();
outputShape = TimesOpOutputShape(m_inputs[0], m_inputs[1], outputRank, inferInputRankToMap, true);
break;
}
case PrimitiveOpType::TransposeTimes:
{
assert(m_inputs.size() == 2);
auto transposeShapeFunc = [](const NDShape& shape) {
NDShape transposedShape(std::max<size_t>(2, shape.Rank()), 1);
for (size_t i = 0; i < shape.Rank(); ++i)
transposedShape[transposedShape.Rank() - i - 1] = shape[i];
return transposedShape;
};
if (m_inputs[0].Shape().Rank() > 2)
LogicError("Function '%S': TransposeTimes operation currently requires the %s operand '%S' to be of rank 1 or 2", AsString().c_str(), Internal::IsReversingTensorShapesInErrorMessagesEnabled() ? "right" : "left", m_inputs[0].AsString().c_str());
NDShape transposedLeftOperandShape = transposeShapeFunc(m_inputs[0].Shape());
Variable dummyLeftOperand = PlaceholderVariable(transposedLeftOperandShape);
size_t outputRank = m_attributes[PrimitiveFunction::AttributeNameOutputRank].Value<size_t>();
outputShape = TimesOpOutputShape(dummyLeftOperand, m_inputs[1], outputRank, -1, true);
if (dummyLeftOperand.Shape() != transposedLeftOperandShape)
m_inputs[0].m_dataFields->m_shape = transposeShapeFunc(dummyLeftOperand.Shape());
break;
}
case PrimitiveOpType::Convolution:
{
assert(m_inputs.size() == 2);
auto& strides = m_attributes[PrimitiveFunction::AttributeNameStrides].Value<NDShape>();
auto& lowerPad = m_attributes[PrimitiveFunction::AttributeNameLowerPad].Value<NDShape>();
auto& upperPad = m_attributes[PrimitiveFunction::AttributeNameUpperPad].Value<NDShape>();
NDShape tmpShape = NDShape::Unknown;
if (m_attributes.Contains(PrimitiveFunction::AttributeNameOutputShape))
tmpShape = m_attributes[PrimitiveFunction::AttributeNameOutputShape].Value<NDShape>();
auto sharing = AsVector<bool>(m_attributes[PrimitiveFunction::AttributeNameSharing].Value<std::vector<DictionaryValue>>());
auto autoPadding = AsVector<bool>(m_attributes[PrimitiveFunction::AttributeNameAutoPadding].Value<std::vector<DictionaryValue>>());
bool transpose = m_attributes[PrimitiveFunction::AttributeNameTranspose].Value<bool>();
if (m_inputs[0].Shape().Rank() < m_inputs[1].Shape().Rank())
InvalidArgument("The convolution map operand '%S' rank (%d) should be >= rank (%d) of the shape of the input operand '%S'.",
m_inputs[0].AsString().c_str(), (int)m_inputs[0].Shape().Rank(), (int)m_inputs[1].Shape().Rank(), m_inputs[1].AsString().c_str());
NDShape outputMapCount, kernelShape;
std::tie(outputMapCount, kernelShape) = GetConvolutionOutputMapCountAndKernelShape(m_inputs[0].Shape(), m_inputs[1].Shape(), transpose);
auto originalKernelShape = kernelShape;
auto inputShape = m_inputs[1].Shape();
if (!transpose || tmpShape.IsUnknown() || tmpShape[0] == 0)
{
outputShape = ConvolutionOpOutputShape(m_op, inputShape, kernelShape, outputMapCount, strides, sharing, autoPadding, lowerPad, upperPad, transpose, true);
}
else
{
NDShape inferredInputShape = ConvolutionOpOutputShape(m_op, tmpShape, kernelShape, outputMapCount, strides, sharing, autoPadding, lowerPad, upperPad, false, true);
if (inferredInputShape != inputShape)
{
RuntimeError("Convolution transpose: The shape '%S' of the convolution transpose operand '%S' is different than the resulting shape '%S' from convolving the "
"specified output shape '%S' using the provided options.",
inputShape.AsString().c_str(), m_inputs[1].AsString().c_str(), inferredInputShape.AsString().c_str(), tmpShape.AsString().c_str());
}
outputShape = tmpShape;
}
auto kernelRank = kernelShape.Rank();
if (originalKernelShape != kernelShape)
{
for (size_t i2 = 0; i2 < kernelRank; ++i2)
m_inputs[0].m_dataFields->m_shape[i2] = kernelShape[i2];
}
if (transpose && (m_inputs[0].Shape().Rank() > kernelRank) && (m_inputs[0].Shape()[kernelRank] == NDShape::InferredDimension))
m_inputs[0].m_dataFields->m_shape[kernelRank] = outputMapCount[outputMapCount.Rank()-1];
m_attributes[PrimitiveFunction::AttributeNameSharing] = AsDictionaryValueVector(sharing);
m_attributes[PrimitiveFunction::AttributeNameAutoPadding] = AsDictionaryValueVector(autoPadding);
break;
}
case PrimitiveOpType::CrossEntropyWithSoftmax:
case PrimitiveOpType::Logistic:
case PrimitiveOpType::LambdaRank:
case PrimitiveOpType::CosDistance:
case PrimitiveOpType::SquaredError:
case PrimitiveOpType::EditDistanceError:
case PrimitiveOpType::ClassificationError:
case PrimitiveOpType::NDCG:
{
if ((m_op == PrimitiveOpType::ClassificationError) || (m_op == PrimitiveOpType::Logistic))
assert(m_inputs.size() >= 2);
else if ((m_op == PrimitiveOpType::LambdaRank) || (m_op == PrimitiveOpType::NDCG))
assert(m_inputs.size() == 3);
else
assert(m_inputs.size() == 2);
// Validate that the first 2 operands are elementwise compatible and also infer operand shapes as needed
BinaryElementwiseOpOutputShape(m_op, m_inputs[0], m_inputs[1], true, true);
if (m_op == PrimitiveOpType::ClassificationError)
{
if ((m_inputs.size() == 3) && !IsConstantScalar(m_inputs[2]))
InvalidArgument("ClassificationError: Input(2) '%S' correponds to topK input and must be a scalar constant.", m_inputs[2].AsString().c_str());
}
else if (m_op == PrimitiveOpType::Logistic)
{
if (m_inputs.size() == 3)
BinaryElementwiseOpOutputShape(m_op, m_inputs[0], m_inputs[2], true, true);
}
outputShape = {};
break;
}
case PrimitiveOpType::ForwardBackward:
{
assert(m_inputs.size() == 2);
if (m_inputs[0].Shape().TotalSize() != m_inputs[1].Shape().TotalSize())
InvalidArgument("ForwardBackward: The shapes of operands '%S' and '%S' must have the same total size.", m_inputs[0].AsString().c_str(), m_inputs[1].AsString().c_str());
outputShape = {};
break;
}
case PrimitiveOpType::ReduceElements:
{
assert(m_inputs.size() == 1);
auto reductionAxis = NormalizeAxis(m_attributes[PrimitiveFunction::AttributeNameAxis].Value<Axis>(), m_inputs[0]);
if (reductionAxis == Axis::AllStaticAxes() || reductionAxis == Axis::AllAxes())
outputShape = {};
else if (reductionAxis.IsDynamicAxis())
outputShape = m_inputs[0].Shape();
else
{
std::vector<int> reductionAxes = { reductionAxis.StaticAxisIndex() };
outputShape = ReductionOpOutputShape(m_op, m_inputs[0].Shape(), reductionAxes, /*preserveReductionAxes =*/ true);
}
break;
}
case PrimitiveOpType::BatchNormalization:
{
assert(m_inputs.size() == 6);
auto spatial = m_attributes[PrimitiveFunction::AttributeNameSpatial].Value<bool>();
outputShape = BatchNormalizationOutputShape(m_inputs, spatial, true);
break;
}
case PrimitiveOpType::GatherPacked:
{
bool sourceHasDynamicAxis = !m_inputs[0].DynamicAxes().empty();
// inherit tensor dimension from sourceData, minus the last (column or time) dimension. TODO this needs to become simpler...
if (sourceHasDynamicAxis)
outputShape = m_inputs[0].Shape();
else
{
if (m_inputs[0].Shape().Rank() > 1)
outputShape = outputShape.SubShape(0, outputShape.Rank() - 1);
else
outputShape = {};
}
break;
}
case PrimitiveOpType::Splice:
{
assert(m_inputs.size() >= 2);
auto maxInputRank = MaxInputRank(m_inputs);
auto spliceAxis = NormalizeStaticAxis(m_attributes[PrimitiveFunction::AttributeNameAxis].Value<Axis>(), NDShape(maxInputRank));
if (!spliceAxis.IsStaticAxis())
LogicError("Function '%S': Splice operation currently does not support splicing along dynamic axis", AsString().c_str());
if (spliceAxis.StaticAxisIndex() < 0)
InvalidArgument("Function '%S': Splice operation's axis index (%d) must be >= 0.", AsString().c_str(), spliceAxis.StaticAxisIndex());
outputShape = SpliceOutputShape(m_inputs, spliceAxis.StaticAxisIndex());
break;
}
case PrimitiveOpType::RandomSample:
case PrimitiveOpType::RandomSampleInclusionFrequency:
{
auto numSamples = m_attributes[PrimitiveFunction::AttributeNameNumSamples].Value<size_t>();
auto allowDuplicates = m_attributes[PrimitiveFunction::AttributeNameAllowDuplicates].Value<bool>();
if (numSamples == 0)
InvalidArgument("RandomSample/RandomSampleInclusionFrequency: Number of requested samples must be > 0.");
let& shape = m_inputs[0].Shape();
size_t numClasses = shape.Dimensions()[0];
if (numClasses != NDShape::InferredDimension && !allowDuplicates && numClasses <= numSamples)
InvalidArgument("RandomSample/RandomSampleInclusionFrequency: For sampling without duplicates the number of requested samples "
"(%lu) must be less than the number of classes (%lu).", numSamples, numClasses);
// within this block we handle RandomSample and RandomSampleInclusionFrequency
if (m_op == PrimitiveOpType::RandomSampleInclusionFrequency)
outputShape = shape;
else
{
vector<size_t> dimensions{ numClasses, numSamples };
outputShape = NDShape(dimensions);
}
break;
}
case PrimitiveOpType::OptimizedRNNStack:
{
assert(m_inputs.size() == 2);
auto operand = m_inputs[0];
auto parameter = m_inputs[1];
if (operand.Shape().Rank() != 1)
InvalidArgument("OptimizedRNNStack: input '%S' must have rank 1; actual input rank is %lu.", operand.AsString().c_str(), operand.Shape().Rank());
if (operand.DynamicAxes().empty())
InvalidArgument("OptimizedRNNStack: input '%S' must have at least one dynamic axis.", operand.AsString().c_str());
auto numLayers = m_attributes[PrimitiveFunction::AttributeNameNumLayers].Value<size_t>();
if (numLayers == 0)
InvalidArgument("Number of layers (%d) in OptimizedRNNStack operation must be > 0.", (int)numLayers);
auto bidirectional = m_attributes[PrimitiveFunction::AttributeNameBidirectional].Value<bool>();
auto hiddenSize = m_attributes[PrimitiveFunction::AttributeNameHiddenSize].Value<size_t>();
// output dims
outputShape = operand.Shape();
outputShape[0] = (bidirectional ? 2 : 1) * hiddenSize;
// infer input size
// Note: Output dim is second axis, so say initOutputRank=-1.
if (parameter.Shape().Rank() == 2)
{
const auto recurrentOp = m_attributes[PrimitiveFunction::AttributeNameRecurrentOp].Value<std::wstring>();
const auto attributes = RnnAttributes(bidirectional, numLayers, hiddenSize, recurrentOp, -1);
const auto numParameters = attributes.GetNumParameters(operand.Shape().TotalSize());
std::vector<std::pair<Variable, NDShape>> newOperandShapes = { { parameter, std::move(NDShape({ numParameters.first, numParameters.second })) } };
UpdateOperandShapes(newOperandShapes);
}
break;
}
case PrimitiveOpType::ReconcileDynamicAxis:
{
assert(m_inputs.size() == 2);
auto operand = m_inputs[0];
auto layout = m_inputs[1];
// data operand can be a constant or a param matrix
if (layout.DynamicAxes().empty())
InvalidArgument("ReconcileDynamicAxis: layout operand '%S' must have at least one dynamic axis.", layout.AsString().c_str());
outputShape = operand.Shape();
break;
}
case PrimitiveOpType::CosDistanceWithNegativeSamples:
{
assert(m_inputs.size() == 4);
auto shiftInput = m_inputs[2];
auto numNegativeSamplesInput = m_inputs[3];
if (!IsConstantScalar(shiftInput) || !IsConstantScalar(numNegativeSamplesInput))
InvalidArgument("CosDistanceWithNegativeSamples: Input(2) '%S' and Input(3) '%S' correpond to shift and numNegativeSamples inputs and must be scalar constants.",
shiftInput.AsString().c_str(), numNegativeSamplesInput.AsString().c_str());
auto numNegativeSamples = (size_t)Constant(numNegativeSamplesInput).Value()->AsScalar<float>();
outputShape = NDShape({ numNegativeSamples + 1 });
break;
}
default:
LogicError("Specified Primitive Function op %S is not supported", PrimitiveOpTypeName(m_op).c_str());
break;
}
}
}
}
outputs.push_back({ OutputVariable(outputShape, outputDataType, outputDynamicAxes, needsGradient, Name().empty() ? L"" : Name()) });
}
}
static const std::wstring s_primitiveFunctionTypeValue = L"PrimitiveFunction";
/*virtual*/ Dictionary PrimitiveFunction::Serialize() const
{
Dictionary dict;
dict[versionKey] = CurrentVersion();
dict[typeKey] = s_primitiveFunctionTypeValue;
dict[opKey] = static_cast<size_t>(m_op);
dict[attributesKey] = Attributes();
dict[uidKey] = Uid();
if (!Name().empty())
dict[nameKey] = Name();
auto inputs = Inputs();
vector<DictionaryValue> inputUids;
inputUids.reserve(inputs.size());
for (auto& input : inputs)
{
inputUids.push_back(input.Uid());
}
dict[inputsKey] = std::move(inputUids);
if (m_op == PrimitiveOpType::Block)
{
auto blockFunction = dynamic_cast<const BlockFunction*>(this);
auto blockCompositeFunc = dynamic_cast<const CompositeFunction*>(blockFunction->Composite().get());
dict[blockFunctionCompositeKey] = blockCompositeFunc->SerializeBlockComposite();
dict[blockFunctionOpNameKey] = OpName();
const auto& blockArgumentsMap = BlockArgumentsMapping();
std::vector<std::wstring> serializedArgumentsMapKeys;
std::vector<std::wstring> serializedArgumentsMapValues;
for (auto argumentMapping : blockArgumentsMap)
{
serializedArgumentsMapKeys.push_back(argumentMapping.first.Uid());
serializedArgumentsMapValues.push_back(argumentMapping.second.Uid());
}
dict[blockFunctionCompositeArgumentsMapKeysKey] = AsDictionaryValueVector(serializedArgumentsMapKeys);
dict[blockFunctionCompositeArgumentsMapValuesKey] = AsDictionaryValueVector(serializedArgumentsMapValues);
}
return dict;
}
/*static*/ FunctionPtr PrimitiveFunction::Deserialize(const Dictionary& dict,
const std::unordered_map<std::wstring, Variable>& uidToVariableMap,
const std::unordered_set<FunctionPtr>& allPrimitiveFunctions,
const std::unordered_map<Variable, Variable>& placeholderReplacements,
const CNTK::DeviceDescriptor& device)
{
static const vector<std::wstring> s_requiredDictionaryKeys = { typeKey, opKey, uidKey, attributesKey, inputsKey };
size_t version = ValidateDictionary<PrimitiveFunction>(dict, s_requiredDictionaryKeys, s_primitiveFunctionTypeValue, s_serializationVersion);
PrimitiveOpType op = PrimitiveOpType(dict[opKey].Value<std::size_t>());
// The hard requirement that the serialization depends on is that
// new op type values are only added to the end of the list.
// This also applies to other enums (DataType, VariableKind, etc.)
if (op >= PrimitiveOpType::UnknownOP)
{
CNTK::LogicError("Unexpected op '%ls':'%u' (%s).",
opKey.c_str(),
static_cast<std::underlying_type<CNTK::PrimitiveOpType>::type>(op),
GetVersionsString<PrimitiveFunction>(s_serializationVersion, version).c_str());
}
const auto& uid = dict[uidKey].Value<std::wstring>();
std::wstring name = L"";
if (dict.Contains(nameKey))
name = dict[nameKey].Value<std::wstring>();
auto attributes = dict[attributesKey].Value<Dictionary>();
const auto& inputUids = dict[inputsKey].Value<vector<DictionaryValue>>();
std::vector<Variable> inputs;
inputs.reserve(inputUids.size());
for (const auto& dictionaryValue : inputUids)
{
const auto& inputUid = dictionaryValue.Value<std::wstring>();
if (uidToVariableMap.find(inputUid) == uidToVariableMap.end())
{
CNTK::LogicError("There are no inputs corresponding to input uid = '%ls' (%s).",
inputUid.c_str(), GetVersionsString<PrimitiveFunction>(s_serializationVersion, version).c_str());
}
inputs.push_back(uidToVariableMap.at(inputUid));
}
if (op == PrimitiveOpType::Block)
{
static const vector<std::wstring> s_requiredBlockFunctionDictionaryKeys = { blockFunctionCompositeKey, blockFunctionOpNameKey, blockFunctionCompositeArgumentsMapKeysKey, blockFunctionCompositeArgumentsMapValuesKey };
ValidateDictionary<PrimitiveFunction>(dict, s_requiredBlockFunctionDictionaryKeys, s_primitiveFunctionTypeValue, s_serializationVersion);
auto composite = CompositeFunction::DeserializeBlockComposite(dict[blockFunctionCompositeKey].Value<Dictionary>(), allPrimitiveFunctions, placeholderReplacements, device);
auto compositeArguments = composite->Arguments();
auto findCompositeArgumentByUid = [&compositeArguments](const std::wstring& uid) {
return *std::find_if(compositeArguments.begin(), compositeArguments.end(), [&uid](const Variable& argument) {
return (argument.Uid() == uid);
});
};
const auto& blockOpName = dict[blockFunctionOpNameKey].Value<std::wstring>();
auto blockArgumentsMapKeys = AsVector<std::wstring>(dict[blockFunctionCompositeArgumentsMapKeysKey].Value<std::vector<DictionaryValue>>());
auto blockArgumentsMapValues = AsVector<std::wstring>(dict[blockFunctionCompositeArgumentsMapValuesKey].Value<std::vector<DictionaryValue>>());
if (blockArgumentsMapKeys.size() != blockArgumentsMapValues.size())
RuntimeError("Invalid block function dictionary found during deserialization; Number (%d) of block argument map keys does not match "
"the number (%d) of map values.", (int)blockArgumentsMapKeys.size(), (int)blockArgumentsMapValues.size());
std::vector<std::pair<Variable, Variable>> argumentsMap;
for (size_t i = 0; i < blockArgumentsMapKeys.size(); ++i)
argumentsMap.push_back({ findCompositeArgumentByUid(blockArgumentsMapKeys[i]), uidToVariableMap.at(blockArgumentsMapValues[i]) });
return std::shared_ptr<BlockFunction>(new BlockFunction(std::move(composite), argumentsMap, blockOpName, std::move(attributes), name, uid),
[](BlockFunction* ptr) { delete ptr; });
}
if (version < 4 && op == PrimitiveOpType::BatchNormalization)
{
if (GetTraceLevel() >= TraceLevel::Warning)
{
// TODO: all logging functionality should be refactored to live in a logging utility class.
fprintf(stderr, "WARNING: the dictionary (version=%zu) does not contain a required "
"BatchNormalization parameter for the running mean sample count. "
"Injected a new parameter with a value of '0'.", version);
}
// patch up old the model by adding an extra input
auto runCount = Constant::Scalar(0.0f, device);
// HACK: uid has to be changed (by adding some unique prefix to the auto-generated "Constant"+ID_counter)
// to avoid conflicts with uids recorded in the function graph, which we are deserializing.
runCount.m_dataFields->m_uid = L"BatchNormSampleCount" + runCount.m_dataFields->m_uid;
inputs.push_back(runCount);
}
return std::shared_ptr<PrimitiveFunction>(new PrimitiveFunction(op, inputs, std::move(attributes), name, uid),
[](PrimitiveFunction* ptr) { delete ptr; });
}
static const vector<wstring> s_stateAttributes = { PrimitiveFunction::AttributeNameRngSeed, PrimitiveFunction::AttributeNameRngOffset };
Dictionary PrimitiveFunction::GetState() const
{
if (!IsStateful())
LogicError("Function '%S' is not stateful.", AsString().c_str());
Dictionary state;
for (auto& key : s_stateAttributes)
{
state[key] = m_attributes[key];
}
return state;
}
void PrimitiveFunction::SetState(const Dictionary& state)
{
if (!IsStateful())
LogicError("Function '%S' is not stateful.", AsString().c_str());
for (auto& key : s_stateAttributes)
{
m_attributes[key] = state[key];
}
}
/*static*/ void PrimitiveFunction::FixNDShape(size_t filterRank, size_t inputRank, NDShape& shape, size_t deflt, const NDShape& from/* = NDShape()*/)
{
auto dims = shape.Dimensions();
Microsoft::MSR::CNTK::ConvolutionNodeBase<float>::FixVectorShape(filterRank, inputRank, dims, deflt, from.Dimensions());
shape = NDShape(dims);
}
NDShape PrimitiveFunction::ConvolutionOpOutputShape(PrimitiveOpType op, const NDShape& operandShape, NDShape& kernelShape, NDShape& outputMapCount, NDShape& strides,
std::vector<bool>& sharing, std::vector<bool>& autoPad, NDShape& lowerPad, NDShape& upperPad,
bool transpose, bool inferDimensions, bool ceilOutputDim/* = false*/) const
{
if (inferDimensions)
{
size_t inputRank = operandShape.Rank();
// Unknown kernel shape valid only for pooling, however, the shape should have expanded before
// this call.
if (kernelShape == NDShape::Unknown)
RuntimeError("Convolution: Kernel shape can't be Unknown.");
// infer reduction dimensions if not given
// If kernel has a lower rank than the input then the remaining dimensions are to be reduced over.
size_t filterRank = kernelShape.Rank();
// If the trailing axis dimensionality of the kernel shape is NDShape::InferredDimension, we reduce over it by
// picking the corresponding operand shape dimensionality
// This is done by shrinking the filter rank and let the dimensions be inferred from the operand's shape
// TODO: Should we do this for all of the axes in kernelShape that have a dimensionailty of NDShape::InferredDimension?
if ((filterRank > 0) && (kernelShape[filterRank - 1] == NDShape::InferredDimension))
{
filterRank--;
kernelShape = kernelShape.SubShape(0, filterRank);
}
NDShape fromShape;
if (op == PrimitiveOpType::Convolution)
fromShape = operandShape;
size_t fillRank = (!transpose) ? filterRank : filterRank - 1;
FixNDShape(fillRank, inputRank, kernelShape, 1, fromShape); // convolve over red dim; pool over 1
FixNDShape(fillRank, inputRank, strides, 1, fromShape); // stride for reduction dims is red dim or 1
FixNDShape(fillRank, inputRank, lowerPad, 0);
FixNDShape(fillRank, inputRank, upperPad, 0);
Microsoft::MSR::CNTK::ConvolutionNodeBase<float>::FixVectorShape(fillRank, inputRank, sharing, true);
Microsoft::MSR::CNTK::ConvolutionNodeBase<float>::FixVectorShape(fillRank, inputRank, autoPad, false); // no padding for reduction dims
}
decltype(&Microsoft::MSR::CNTK::ConvolveGeometry::ComputeOutputShape) computeOutputShapeFunc;
if (!transpose)
computeOutputShapeFunc = &Microsoft::MSR::CNTK::ConvolveGeometry::ComputeOutputShape;
else
computeOutputShapeFunc = &Microsoft::MSR::CNTK::ConvolveGeometry::ComputeInputShape;
return AsNDShape(computeOutputShapeFunc(AsTensorShape(operandShape), AsTensorShape(kernelShape), AsTensorShape(outputMapCount), AsTensorShape(strides), sharing, autoPad, AsTensorShape(lowerPad), AsTensorShape(upperPad), ceilOutputDim));
}
/*static*/ bool PrimitiveFunction::UpdateOperandShapes(std::vector<std::pair<Variable, NDShape>>& newOperandShapes)
{
bool anyParameterOperandDimsInferred = false;
auto updateOperandShapeFunc = [](Variable& operand, const NDShape& newOperandShape) {
if ((operand.IsParameter() || operand.IsConstant()) && (operand.Shape() != newOperandShape))
{
operand.m_dataFields->m_shape = newOperandShape;
return true;
}
return false;
};
for (auto& newOperandShapePair : newOperandShapes)
anyParameterOperandDimsInferred = updateOperandShapeFunc(newOperandShapePair.first, newOperandShapePair.second) || anyParameterOperandDimsInferred;
return anyParameterOperandDimsInferred;
}
NDShape PrimitiveFunction::NaryElementwiseOpOutputShape(PrimitiveOpType op, std::vector<Variable>& operands, bool broadcastAllowed, bool inferInputDimensions) const
{
assert(operands.size() > 1);
// TODO: Is this logic of transitively constructing the output shape from the operands correct?
Variable dummyOutputVariable = PlaceholderVariable(NDShape());
for (auto& operand : operands)
dummyOutputVariable.m_dataFields->m_shape = BinaryElementwiseOpOutputShape(op, dummyOutputVariable, operand, broadcastAllowed, inferInputDimensions);
return dummyOutputVariable.Shape();
}
}