parameters: (parallel_io=False, num_io_partitions=1)
return_statement: return {
          'DISTCONV_WS_CAPACITY_FACTOR': 0.8,
          'DISTCONV_OVERLAP_HALO_EXCHANGE': 1,
          'LBANN_DISTCONV_HALO_EXCHANGE': 'HYBRID',
          'LBANN_DISTCONV_TENSOR_SHUFFLER': 'HYBRID',
          'LBANN_DISTCONV_CONVOLUTION_FWD_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_CONVOLUTION_BWD_DATA_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_CONVOLUTION_BWD_FILTER_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_RANK_STRIDE': 1,
          'LBANN_DISTCONV_COSMOFLOW_PARALLEL_IO': parallel_io,
          'LBANN_DISTCONV_NUM_IO_PARTITIONS': num_io_partitions,
      }
docstring: Return recommended Distconv variables.
  
      Args:
          parallel_io (bool):
              Whether to read a single sample in parallel.
          num_io_partitions (int):
              The number of processes to read a single sample.
docstring_summary: Return recommended Distconv variables.
docstring_tokens: Return recommended Distconv variables .
function: def get_distconv_environment(parallel_io=False, num_io_partitions=1):
      """Return recommended Distconv variables.
  
      Args:
          parallel_io (bool):
              Whether to read a single sample in parallel.
          num_io_partitions (int):
              The number of processes to read a single sample.
      """
  
      return {
          'DISTCONV_WS_CAPACITY_FACTOR': 0.8,
          'DISTCONV_OVERLAP_HALO_EXCHANGE': 1,
          'LBANN_DISTCONV_HALO_EXCHANGE': 'HYBRID',
          'LBANN_DISTCONV_TENSOR_SHUFFLER': 'HYBRID',
          'LBANN_DISTCONV_CONVOLUTION_FWD_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_CONVOLUTION_BWD_DATA_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_CONVOLUTION_BWD_FILTER_ALGORITHM': 'AUTOTUNE',
          'LBANN_DISTCONV_RANK_STRIDE': 1,
          'LBANN_DISTCONV_COSMOFLOW_PARALLEL_IO': parallel_io,
          'LBANN_DISTCONV_NUM_IO_PARTITIONS': num_io_partitions,
      }
function_tokens: def get_distconv_environment ( parallel_io = False , num_io_partitions = 1 ) : return { 'DISTCONV_WS_CAPACITY_FACTOR' : 0.8 , 'DISTCONV_OVERLAP_HALO_EXCHANGE' : 1 , 'LBANN_DISTCONV_HALO_EXCHANGE' : 'HYBRID' , 'LBANN_DISTCONV_TENSOR_SHUFFLER' : 'HYBRID' , 'LBANN_DISTCONV_CONVOLUTION_FWD_ALGORITHM' : 'AUTOTUNE' , 'LBANN_DISTCONV_CONVOLUTION_BWD_DATA_ALGORITHM' : 'AUTOTUNE' , 'LBANN_DISTCONV_CONVOLUTION_BWD_FILTER_ALGORITHM' : 'AUTOTUNE' , 'LBANN_DISTCONV_RANK_STRIDE' : 1 , 'LBANN_DISTCONV_COSMOFLOW_PARALLEL_IO' : parallel_io , 'LBANN_DISTCONV_NUM_IO_PARTITIONS' : num_io_partitions , }