gangliao-patch-1
hedaoyuan 8 years ago
parent 3b65bc7a26
commit b6de52c47b

@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and See the License for the specific language governing permissions and
limitations under the License. */ limitations under the License. */
#include "ConvFunc.h" #include "ConvOp.h"
namespace paddle { namespace paddle {
@ -102,10 +102,10 @@ public:
size_t outputHeight = outputs[0].shape()[2]; size_t outputHeight = outputs[0].shape()[2];
size_t outputWidth = outputs[0].shape()[3]; size_t outputWidth = outputs[0].shape()[3];
float* inputData = inputs[0].data<float>(); real* inputData = inputs[0].data<real>();
float* filterData = inputs[1].data<float>(); real* filterData = inputs[1].data<real>();
float* outputData = outputs[0].data<float>(); real* outputData = outputs[0].data<real>();
NaiveConvFunctor<float> conv; NaiveConvFunctor<real> conv;
conv(inputData, conv(inputData,
batchSize, batchSize,
inputChannels, inputChannels,
Loading…
Cancel
Save