(
attrs: {[key: string]: tensorflow.IAttrValue}, name: string,
def: number[])
| 563 | } |
| 564 | |
| 565 | export function getNumericArrayParam( |
| 566 | attrs: {[key: string]: tensorflow.IAttrValue}, name: string, |
| 567 | def: number[]): number[] { |
| 568 | const param = attrs[name]; |
| 569 | if (param) { |
| 570 | return ((param.list.f && param.list.f.length ? param.list.f : |
| 571 | param.list.i) || |
| 572 | []) |
| 573 | .map(v => (typeof v === 'number') ? v : parseInt(v, 10)); |
| 574 | } |
| 575 | return def; |
| 576 | } |
| 577 | |
| 578 | export function getStringArrayParam( |
| 579 | attrs: {[key: string]: tensorflow.IAttrValue}, name: string, def: string[], |