op { | |
name: "FFT2D" | |
input_arg { | |
name: "input" | |
type: DT_COMPLEX64 | |
} | |
output_arg { | |
name: "output" | |
type: DT_COMPLEX64 | |
} | |
} | |
op { | |
name: "FFT2D" | |
input_arg { | |
name: "input" | |
type_attr: "Tcomplex" | |
} | |
output_arg { | |
name: "output" | |
type_attr: "Tcomplex" | |
} | |
attr { | |
name: "Tcomplex" | |
type: "type" | |
default_value { | |
type: DT_COMPLEX64 | |
} | |
allowed_values { | |
list { | |
type: DT_COMPLEX64 | |
type: DT_COMPLEX128 | |
} | |
} | |
} | |
} |