(initializer_range=0.02, data_type=tf.float32)
| 245 | self.top_p = top_p |
| 246 | |
| 247 | def create_initializer(initializer_range=0.02, data_type=tf.float32): |
| 248 | return tf.truncated_normal_initializer(stddev=initializer_range, dtype=data_type) |
| 249 | |
| 250 | def _get_shape_invariants(tensor): |
| 251 | """Returns the shape of the tensor but sets middle dims to None.""" |
no outgoing calls
no test coverage detected