"docs/source/en/training/unconditional_training.md" did not exist on "3b641eabe9876e7c48977b35331fda54ce972b4a"
Commit c7aa455f authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Re-format 'DeviceElementwise'

parent 339e51d1
......@@ -260,14 +260,20 @@ struct DeviceElementwise
return IsSupportedArgument(*dynamic_cast<const Argument*>(p_arg));
}
static auto MakeArgument(const std::array<index_t, NumDim> lengths,
static auto
MakeArgument(const std::array<index_t, NumDim> lengths,
const std::array<std::array<index_t, NumDim>, NumInput> inStridesArray,
const std::array<std::array<index_t, NumDim>, NumOutput> outStridesArray,
const std::array<const void*, NumInput> in_dev_buffers,
const std::array<void*, NumOutput> out_dev_buffers,
ElementwiseOperation elementwise_op)
{
return Argument{lengths, inStridesArray, outStridesArray, in_dev_buffers, out_dev_buffers, elementwise_op};
return Argument{lengths,
inStridesArray,
outStridesArray,
in_dev_buffers,
out_dev_buffers,
elementwise_op};
}
std::unique_ptr<BaseArgument>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment