Home
last modified time | relevance | path

Searched refs:DynamicTypeTrait (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/aten/src/ATen/core/
Dtype_factory.h20 c10::DynamicTypeTrait<T>::tagValue(),
27 c10::DynamicTypeTrait<T>::tagValue(),
42 c10::DynamicTypeTrait<T>::tagValue(),
48 return DynamicTypeTrait<T>::getBaseType();
58 std::enable_if_t<DynamicTypeTrait<T>::isBaseType, int> = 0>
66 std::enable_if_t<!DynamicTypeTrait<T>::isBaseType, int> = 0>
Ddynamic_type.cpp334 return DynamicTypeTrait<NoneType>::getBaseType(); in get()
336 return DynamicTypeTrait<TensorType>::getBaseType(); in get()
338 return DynamicTypeTrait<FloatType>::getBaseType(); in get()
340 return DynamicTypeTrait<ComplexType>::getBaseType(); in get()
342 return DynamicTypeTrait<IntType>::getBaseType(); in get()
344 return DynamicTypeTrait<BoolType>::getBaseType(); in get()
346 return DynamicTypeTrait<StringType>::getBaseType(); in get()
354 return DynamicTypeTrait<DeviceObjType>::getBaseType(); in get()
356 return DynamicTypeTrait<StreamObjType>::getBaseType(); in get()
360 return DynamicTypeTrait<CapsuleType>::getBaseType(); in get()
[all …]
Ddynamic_type.h211 struct DynamicTypeTrait { struct
224 struct TORCH_API DynamicTypeTrait<NAME##Type> { \ argument
Dtype_factory.cpp45 {#NAME, c10::DynamicTypeTrait<c10::TYPE>::getBaseType()}, in basePythonTypes()
/external/pytorch/torch/csrc/jit/serialization/
Dunpickler.cpp186 return dyn->tag() == c10::DynamicTypeTrait<T>::tagValue(); in is()