Searched refs:tensor_shape_origin_ (Results 1 – 2 of 2) sorted by relevance
44 buffer << std::endl << std::string("tensor shape origin = " + tensor_shape_origin_.ToString()); in OriginToString()52 tensor_shape_origin_ = tensor_shape; in Init()79 if (tensor_shape_origin_.Init(tensor_shape) != SUCCESS) { in InitFromVector()82 if (Init(device_arrangement_origin_, tensor_map_origin_, tensor_shape_origin_) != SUCCESS) { in InitFromVector()93 if (tensor_map_origin_.GetDimSize() != tensor_shape_origin_.GetDimSize()) { in IsValidTensorLayout()146 tensor_shape_ = tensor_shape_origin_; in RemoveElementEqualToOneInDeviceArrangement()411 Shape tensor_shape(tensor_shape_origin_.array()); in TransferRepeatLayout()
138 Arrangement tensor_shape_origin_; variable