"""Tensorflow Branch."""
# yapf: disable
from garage.tf._functions import (center_advs,
                                  compile_function,
                                  compute_advantages,
                                  concat_tensor_dict_list,
                                  concat_tensor_list,
                                  discounted_returns,
                                  filter_valids,
                                  filter_valids_dict,
                                  flatten_batch,
                                  flatten_batch_dict,
                                  flatten_inputs,
                                  flatten_tensor_variables,
                                  get_target_ops,
                                  graph_inputs,
                                  new_tensor,
                                  new_tensor_like,
                                  pad_tensor,
                                  pad_tensor_dict,
                                  pad_tensor_n,
                                  paths_to_tensors,
                                  positive_advs,
                                  split_tensor_dict_list,
                                  stack_tensor_dict_list)

# yapf: enable

__all__ = [
    'compile_function',
    'get_target_ops',
    'flatten_batch',
    'flatten_batch_dict',
    'filter_valids',
    'filter_valids_dict',
    'graph_inputs',
    'flatten_inputs',
    'flatten_tensor_variables',
    'new_tensor',
    'new_tensor_like',
    'concat_tensor_list',
    'concat_tensor_dict_list',
    'stack_tensor_dict_list',
    'split_tensor_dict_list',
    'pad_tensor',
    'pad_tensor_n',
    'pad_tensor_dict',
    'compute_advantages',
    'center_advs',
    'positive_advs',
    'discounted_returns',
    'paths_to_tensors',
]
