Searched defs:shape_type (Results 1 – 9 of 9) sorted by relevance
23 void TensorListReserve::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()32 void TensorListReserve::set_shape_type(const int64_t shape_type) { in set_shape_type()
46 void TensorListFromTensor::Init(const int64_t element_dtype, const int64_t shape_type) { in Init()65 void TensorListFromTensor::set_shape_type(const int64_t shape_type) { in set_shape_type()
33 little_int32_buf_at shape_type; member
25 def get_broadcast_shape(x_shape, y_shape, prim_name, shape_type=""): argument
129 typedef double shape_type; typedef
70 UnknowShapeOpType shape_type = UnknowShapeOpType::DEPEND_IN_SHAPE; in Initialize() local
288 auto *shape_type = reinterpret_cast<int32_t *>(ext_info_buf + ext_info_offset); in SetExtInfoShapeType() local392 UnknowShapeOpType shape_type = UnknowShapeOpType::DEPEND_IN_SHAPE; in CreateExtInfo() local
173 typedef std::size_t shape_type; typedef in boost::asio::thread_pool::basic_executor_type