public ComputeGraph(IWeightFactory weightFactory, bool needBack = true) { weightMatrixFactory = weightFactory as WeightMatrixFactory; this.needs_backprop = needBack; }
private void InitWeightsFactory() { m_weightFactory = new IWeightFactory[m_deviceIds.Length]; if (m_archType == ArchTypeEnums.GPU_CUDA) { for (int i = 0; i < m_deviceIds.Length; i++) { m_weightFactory[i] = new WeightTensorFactory(); } } else { for (int i = 0; i < m_deviceIds.Length; i++) { m_weightFactory[i] = new WeightMatrixFactory(); } } }