Searched defs:element_dtype (Results 1 – 6 of 6) sorted by relevance
23 void TensorListGetItem::Init(const int64_t element_dtype) { this->set_element_dtype(element_dtype);… in Init()25 void TensorListGetItem::set_element_dtype(const int64_t element_dtype) { in set_element_dtype()
23 void TensorListSetItem::Init(const int64_t element_dtype) { this->set_element_dtype(element_dtype);… in Init()25 void TensorListSetItem::set_element_dtype(const int64_t element_dtype) { in set_element_dtype()
23 void TensorListReserve::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()28 void TensorListReserve::set_element_dtype(const int64_t element_dtype) { in set_element_dtype()
26 void TensorListStack::Init(const int64_t num_elements, const int64_t element_dtype) { in Init()35 void TensorListStack::set_element_dtype(const int64_t element_dtype) { in set_element_dtype()
46 void TensorListFromTensor::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()61 void TensorListFromTensor::set_element_dtype(const int64_t element_dtype) { in set_element_dtype()
336 auto element_dtype = static_cast<size_t>(scratch); in ConvertConstVariant() local