| 64 | } |
| 65 | |
| 66 | size_t CactusGraph::pow(size_t input, float exponent) { |
| 67 | const auto& input_buffer = get_output_buffer(input); |
| 68 | OpParams params{.scalar = exponent, .output_precision = input_buffer.precision}; |
| 69 | return add_node(OpType::POW, {input}, input_buffer.shape, params); |
| 70 | } |
| 71 | |
| 72 | size_t CactusGraph::view(size_t input, const std::vector<size_t>& new_shape) { |
| 73 | const auto& input_buffer = get_output_buffer(input); |