Return a tweaked version of 'base_shape'.
(parameters, delta)
| 2214 | }] |
| 2215 | |
| 2216 | def get_shape(parameters, delta): |
| 2217 | """Return a tweaked version of 'base_shape'.""" |
| 2218 | axis = parameters["axis"] |
| 2219 | shape = parameters["base_shape"][:] |
| 2220 | if axis < 0: |
| 2221 | axis += len(shape) |
| 2222 | if axis < len(shape): |
| 2223 | shape[axis] += delta |
| 2224 | return shape |
| 2225 | |
| 2226 | def build_graph(parameters): |
| 2227 | all_tensors = [] |
no outgoing calls
no test coverage detected