parent
f79207033b
commit
5a34ccf3d4
|
@ -38,7 +38,7 @@ namespace sd {
|
||||||
auto dtype = block.numD() ? D_ARG(0) : ArrayOptions::dataType(in);
|
auto dtype = block.numD() ? D_ARG(0) : ArrayOptions::dataType(in);
|
||||||
auto shape = sd::ConstantShapeHelper::getInstance()->createShapeInfo(dtype, in);
|
auto shape = sd::ConstantShapeHelper::getInstance()->createShapeInfo(dtype, in);
|
||||||
|
|
||||||
nd4j_printf("numD: %i; dtype: %s\n", block.numD(), DataTypeUtils::asString(dtype).c_str());
|
//nd4j_printf("numD: %i; dtype: %s\n", block.numD(), DataTypeUtils::asString(dtype).c_str());
|
||||||
|
|
||||||
return SHAPELIST(shape);
|
return SHAPELIST(shape);
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue