TensorIteratorConfig |
TensorIteratorConfig.add_borrowed_const_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_borrowed_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_borrowed_output(TensorBase output) |
TensorIteratorConfig |
TensorIteratorConfig.add_const_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_output(TensorBase output)
Construction
|
TensorIteratorConfig |
TensorIteratorConfig.add_owned_const_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_owned_input(TensorBase input) |
TensorIteratorConfig |
TensorIteratorConfig.add_owned_output(TensorBase output) |
TensorIteratorConfig |
TensorIteratorConfig.allow_cpu_scalars(boolean _allow_cpu_scalars) |
TensorIteratorConfig |
TensorIteratorConfig.cast_common_dtype_to_outputs(boolean _cast_common_dtype_to_outputs) |
TensorIteratorConfig |
TensorIteratorConfig.check_all_same_device(boolean _check_all_same_device) |
TensorIteratorConfig |
TensorIteratorConfig.check_all_same_dtype(boolean _check_all_same_dtype) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_device(Device device) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_dtype_and_device(torch.ScalarType dtype,
Device device) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_dtype(torch.ScalarType dtype) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_shape(long... shape) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_shape(long[] shape,
long... squash_dims) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_shape(LongArrayRef shape) |
TensorIteratorConfig |
TensorIteratorConfig.declare_static_shape(LongArrayRef shape,
LongArrayRef squash_dims) |
TensorIteratorConfig |
TensorIteratorConfig.enforce_linear_iteration() |
TensorIteratorConfig |
TensorIteratorConfig.enforce_linear_iteration(boolean _enforce_linear_iteration) |
TensorIteratorConfig |
TensorIteratorConfig.enforce_safe_casting_to_output(boolean _enforce_safe_casting_to_output) |
TensorIteratorConfig |
TensorIteratorConfig.getPointer(long i) |
TensorIteratorConfig |
TensorIteratorConfig.is_reduction(boolean _is_reduction) |
TensorIteratorConfig |
TensorIteratorConfig.position(long position) |
TensorIteratorConfig |
TensorIteratorConfig.promote_inputs_to_common_dtype(boolean _promote_inputs_to_common_dtype) |
TensorIteratorConfig |
TensorIteratorConfig.promote_integer_inputs_to_float(boolean _promote_integer_inputs_to_float) |
TensorIteratorConfig |
TensorIteratorConfig.resize_outputs(boolean resize_outputs) |
TensorIteratorConfig |
TensorIteratorConfig.set_check_mem_overlap(boolean check_mem_overlap) |