b0728e31f312bc703ab227b6ff69f0a92b3dba07,open_seq2seq/utils/utils.py,,cast_types,#Any#Any#,359

Before Change

        if value.dtype.base_dtype != dtype.base_dtype:
          cast_input_dict[key] = tf.cast(value, dtype)
          continue
    if type(value) == dict:
      cast_input_dict[key] = cast_types(input_dict[key], dtype)
      continue
    cast_input_dict[key] = input_dict[key]
  return cast_input_dict

After Change

              cur_list.append(tf.cast(nest_value, dtype))
              continue
        cur_list.append(nest_value)
      cast_input_dict[key] = cur_list
      continue
    cast_input_dict[key] = input_dict[key]
  return cast_input_dict

Instances