Home
last modified time | relevance | path

Searched refs:FastSetupType (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/
DTensorIterator.cpp1353 FastSetupType setup_type = compute_fast_setup_type(config); in fast_set_up()
1354 if (setup_type == FastSetupType::NONE) { in fast_set_up()
1360 case FastSetupType::CONTIGUOUS: in fast_set_up()
1371 case FastSetupType::CHANNELS_LAST: in fast_set_up()
1382 case FastSetupType::NON_OVERLAPPING_DENSE: in fast_set_up()
1420 FastSetupType TensorIteratorBase::compute_fast_setup_type(const TensorIteratorConfig& config) { in compute_fast_setup_type()
1422 return FastSetupType::NONE; in compute_fast_setup_type()
1432 return FastSetupType::NONE; in compute_fast_setup_type()
1436 return FastSetupType::CONTIGUOUS; in compute_fast_setup_type()
1451 return FastSetupType::CONTIGUOUS; in compute_fast_setup_type()
[all …]
DTensorIterator.h238 enum class FastSetupType : uint8_t { enum
633 FastSetupType compute_fast_setup_type(const TensorIteratorConfig&);