parent
f79207033b
commit
5a34ccf3d4
|
@ -38,7 +38,7 @@ namespace sd {
|
|||
auto dtype = block.numD() ? D_ARG(0) : ArrayOptions::dataType(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);
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue