Searched refs:set_element_shape (Results 1 – 11 of 11) sorted by relevance
32 this->set_element_shape(element_shape); in Init()44 void TensorArray::set_element_shape(const std::vector<int> &element_shape) { in set_element_shape() function in mindspore::ops::TensorArray
49 void set_element_shape(const std::vector<int> &element_shape);
102 input0_->set_element_shape(input2_->shape()); in Run()103 output0_->set_element_shape(input2_->shape()); in Run()
53 … output->set_element_shape(std::vector<int>(ele_shape_ptr, ele_shape_ptr + input0->ElementsNum())); in Run()
70 void set_element_shape(const std::vector<int> &shape) { element_shape_ = shape; } in set_element_shape() function
717 input_tensorlist->set_element_shape(input_data_tensorlist->element_shape()); in SetInputShape()
1713 dst_tensor->set_element_shape(src_tensor->element_shape()); in CopyTensorList()
102 dst_tensorlist->set_element_shape(src_tensorlist->element_shape()); in MoveTensorListData()
170 tensor_array->set_element_shape(element_shape); in Process()
130 lite_tensor_list->set_element_shape(element_shape); in ConvertTensorList()293 tensor_list->set_element_shape(tensor_list->tensors().front()->shape()); in SetDataType()
152 …dst->set_element_shape(std::vector<int>(src->element_shape_, src->element_shape_ + src->element_sh… in TensorListC2TensorList()